1580158
/
sd-webui-animatediff
mirror of https://github.com/continue-revolution/sd-webui-animatediff.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Model list fixes (#526) 

* make intermediate directories and handle existing dirs

* fix rm tags for SD2

* return sorted list of models

---------

Co-authored-by: zappityzap <zappityzap@proton.me>
pull/530/head
zappityzap 2024-05-25 21:40:21 -07:00 committed by GitHub
parent 3b7765dd21
commit 0434d99284
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
scripts/animatediff_ui.py
View File

@ -259,17 +259,20 @@ class AnimateDiffUiGroup:
def get_model_list(self):
model_dir = motion_module.get_model_dir()
if not os.path.isdir(model_dir):
os.mkdir(model_dir)
os.makedirs(model_dir, exist_ok=True)
def get_sd_rm_tag():
if shared.sd_model.is_sdxl:
return ["sd1"]
elif shared.sd_model.is_sd2:
return ["sd1, xl"]
return ["sd1", "xl"]
elif shared.sd_model.is_sd1:
return ["xl"]
else:
return []
return [f for f in os.listdir(model_dir) if f != ".gitkeep" and not any(tag in f for tag in get_sd_rm_tag())]
return [
f for f in sorted(os.listdir(model_dir))
if f != ".gitkeep" and not any(tag in f for tag in get_sd_rm_tag())
]
def refresh_models(self, *inputs):