model tabular fix

huggingface · Jun 26, 2024 · 01673f1 · 01673f1
1 parent 475336c
commit 01673f1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
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.8.0.dev0"
+__version__ = "0.8.0"
 
 
 def is_colab():

diff --git a/src/autotrain/app/params.py b/src/autotrain/app/params.py
@@ -359,6 +359,7 @@ def _munge_params_img_obj_det(self):
 
     def _munge_params_tabular(self):
         _params = self._munge_common_params()
+        _params["model"] = self.base_model
         if not self.using_hub_dataset:
             _params["id_column"] = "autotrain_id"
             _params["valid_split"] = "validation"