extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-06-19 17:39:16 +02:00
Code Issues Packages Projects Releases Wiki Activity

Bring back the earlier model scanning logic

Browse Source
This commit is contained in:
cmdr2 2024-10-09 18:16:15 +05:30
parent e32b34e2f4
commit 78dcc7cb03
1 changed files with 2 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
ui/easydiffusion/model_manager.py
View File

@ -438,17 +438,12 @@ def getModels(scan_for_malicious: bool = True):
if not os.path.exists(models_dirs[0]): if not os.path.exists(models_dirs[0]):
os.makedirs(models_dirs[0]) os.makedirs(models_dirs[0])
models["options"][model_type] = []
default_tree = models["options"].get(model_type, [])
for model_dir in models_dirs: for model_dir in models_dirs:
try: try:
scanned_models = scan_directory( default_tree = models["options"].get(model_type, [])
models["options"][model_type] = scan_directory(
model_dir, model_extensions, default_entries=default_tree, nameFilter=nameFilter model_dir, model_extensions, default_entries=default_tree, nameFilter=nameFilter
) )
for m in scanned_models:
if m not in models["options"][model_type]:
models["options"][model_type].append(m)
except MaliciousModelException as e: except MaliciousModelException as e:
models["scan-error"] = str(e) models["scan-error"] = str(e)