From 3813e3c2726ddfe9aef0f7610cb27a5a23f96f32 Mon Sep 17 00:00:00 2001 From: Abhishek Thakur Date: Mon, 17 Jun 2024 13:56:24 +0200 Subject: [PATCH] fixes --- src/autotrain/__init__.py | 2 +- src/autotrain/app/colab.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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: