diff --git a/kohya_gui/dreambooth_gui.py b/kohya_gui/dreambooth_gui.py
index c1b2283d2..7a6525490 100644
--- 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
index 50e30acb2..28d26a458 100644
--- 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
index 81bb36b12..fbd176f3a 100644
--- 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
     
     #