fixes

huggingface · Jun 17, 2024 · 3813e3c · 3813e3c
1 parent 9dafb78
commit 3813e3c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/autotrain/__init__.py b/src/autotrain/__init__.py
@@ -41,7 +41,7 @@
 warnings.filterwarnings("ignore", category=UserWarning, module="huggingface_hub")
 
 logger = Logger().get_logger()
-__version__ = "0.7.122"
+__version__ = "0.7.123"
 
 
 def is_colab():

diff --git a/src/autotrain/app/colab.py b/src/autotrain/app/colab.py
@@ -339,7 +339,7 @@ def start_training(b):
             params_val = json.loads(parameters.value)
             if task_dropdown.value.startswith("llm") or task_dropdown.value.startswith("sentence-transformers"):
                 params_val["trainer"] = task_dropdown.value.split(":")[1]
-                params_val = {k: v for k, v in params_val.items() if k != "trainer"}
+                # params_val = {k: v for k, v in params_val.items() if k != "trainer"}
 
             chat_template = params_val.get("chat_template")
             if chat_template is not None: