fix: vae path validation

bmaltais · bmaltais · May 7, 2024 · Apr 19, 2024 · Apr 20, 2024 · Apr 21, 2024
commit 3902707d9b3811549940b43772a5cfb45edfd7d0
diff --git a/kohya_gui/dreambooth_gui.py b/kohya_gui/dreambooth_gui.py
@@ -541,7 +541,7 @@ def train_model(
     if not validate_folder_path(train_data_dir):
         return TRAIN_BUTTON_VISIBLE
 
-    if not validate_file_path(vae):
+    if not validate_model_path(vae):
         return TRAIN_BUTTON_VISIBLE
     #
     # End of path validation

diff --git a/kohya_gui/lora_gui.py b/kohya_gui/lora_gui.py
@@ -732,7 +732,7 @@ def train_model(
     if not validate_folder_path(train_data_dir):
         return TRAIN_BUTTON_VISIBLE
 
-    if not validate_file_path(vae):
+    if not validate_model_path(vae):
         return TRAIN_BUTTON_VISIBLE
 
     #

diff --git a/kohya_gui/textual_inversion_gui.py b/kohya_gui/textual_inversion_gui.py
@@ -542,7 +542,7 @@ def train_model(
     if not validate_folder_path(train_data_dir):
         return TRAIN_BUTTON_VISIBLE
 
-    if not validate_file_path(vae):
+    if not validate_model_path(vae):
         return TRAIN_BUTTON_VISIBLE
 
     #