Update model validation code

bmaltais · May 7, 2024 · 5984c87 · 5984c87
1 parent 39293d3
commit 5984c87
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/kohya_gui/common_gui.py b/kohya_gui/common_gui.py
@@ -1420,10 +1420,8 @@ def validate_model_path(pretrained_model_name_or_path: str) -> bool:
         log.info(f"{msg} SUCCESS")
     else:
         # If not one of the default models, check if it's a valid local path
-        if not os.path.exists(pretrained_model_name_or_path):
-            log.error(f"{msg} FAILED: is missing or does not exist")
+        if not validate_file_path(pretrained_model_name_or_path):
             return False
-        log.info(f"{msg} SUCCESS")
 
     return True