extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-04-24 03:18:29 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix a bug where the .vae.pt extension wouldn't get picked up. Thanks Madrang, rbertus2000 and JeLuf

Browse Source
This commit is contained in:
cmdr2 2023-01-16 18:30:22 +05:30
parent 8ee4364065
commit 5089ac5ad1
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
ui/easydiffusion/model_manager.py
View File

@ -190,7 +190,11 @@ def getModels():
nonlocal models_scanned nonlocal models_scanned
tree = [] tree = []
for entry in os.scandir(directory): for entry in os.scandir(directory):
if entry.is_file() and True in [entry.name.endswith(s) for s in suffixes]: if entry.is_file():
matching_suffix = list(filter(lambda s: entry.name.endswith(s), suffixes))
if len(matching_suffix) == 0: continue
matching_suffix = matching_suffix[0]
mtime = entry.stat().st_mtime mtime = entry.stat().st_mtime
mod_time = known_models[entry.path] if entry.path in known_models else -1 mod_time = known_models[entry.path] if entry.path in known_models else -1
if mod_time != mtime: if mod_time != mtime:
@ -198,7 +202,7 @@ def getModels():
if is_malicious_model(entry.path): if is_malicious_model(entry.path):
raise MaliciousModelException(entry.path) raise MaliciousModelException(entry.path)
known_models[entry.path] = mtime known_models[entry.path] = mtime
tree.append(entry.name.rsplit('.',1)[0]) tree.append(entry.name[:-len(matching_suffix)])
elif entry.is_dir(): elif entry.is_dir():
scan=scan_directory(entry.path, suffixes) scan=scan_directory(entry.path, suffixes)
if len(scan) != 0: if len(scan) != 0: