From 3902707d9b3811549940b43772a5cfb45edfd7d0 Mon Sep 17 00:00:00 2001 From: Krithic Kumar Date: Tue, 7 May 2024 13:41:24 +0530 Subject: [PATCH] fix: vae path validation --- kohya_gui/dreambooth_gui.py | 2 +- kohya_gui/lora_gui.py | 2 +- kohya_gui/textual_inversion_gui.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) 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 #