extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-01-23 06:39:50 +01: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]):
os.makedirs(models_dirs[0])
models["options"][model_type] = []
default_tree = models["options"].get(model_type, [])
for model_dir in models_dirs:
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
)
for m in scanned_models:
if m not in models["options"][model_type]:
models["options"][model_type].append(m)
except MaliciousModelException as e:
models["scan-error"] = str(e)