small fix on training

SUC-DriverOld · Sep 24, 2024 · 1509c41 · 1509c41
1 parent 8ad101d
commit 1509c41
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tools/webUI_for_clouds/webUI_for_clouds.py b/tools/webUI_for_clouds/webUI_for_clouds.py
@@ -746,7 +746,7 @@ def start_training(train_model_type, train_config_path, train_dataset_type, trai
         train_file = "train_accelerate.py"
     else:
         train_file = "train.py"
-    if model_type not in MODEL_TYPE:
+    if train_model_type not in MODEL_TYPE:
         return i18n("模型类型错误, 请重新选择")
     if not os.path.isfile(train_config_path):
         return i18n("配置文件不存在, 请重新选择")

diff --git a/webUI.py b/webUI.py
@@ -1156,7 +1156,7 @@ def start_training(train_model_type, train_config_path, train_dataset_type, trai
         train_file = "train_accelerate.py"
     else:
         train_file = "train.py"
-    if model_type not in MODEL_TYPE:
+    if train_model_type not in MODEL_TYPE:
         return i18n("模型类型错误, 请重新选择")
     if not os.path.isfile(train_config_path):
         return i18n("配置文件不存在, 请重新选择")