diff --git a/src/autotrain/__init__.py b/src/autotrain/__init__.py index f6ac3d4cff..f99d47268a 100644 --- 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 index f1b04e8d1c..31083219b3 100644 --- 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: