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

fixup dirs not being shown due to wrong isdir call

pull/186/head
kabachuha 2023-07-02 19:18:27 +03:00
parent f58716e0d9
commit d16b0f5473
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
scripts/t2v_helpers/args.py
View File

@ -87,12 +87,13 @@ def setup_text2video_settings_dictionary():
models = []
if os.path.isdir(os.path.join(ph.models_path, 'ModelScope/t2v')):
models.append('<modelscope>')
if os.path.isdir(os.path.join(ph.models_path, 'VideoCrafter')):
if os.path.isdir(os.path.join(ph.models_path, 'VideoCrafter/')):
models.append('<videocrafter>')
models_dir = os.path.join(ph.models_path, 'text2video/')
for subdir in os.listdir(models_dir):
if os.path.isdir(subdir):
models.append(subdir)
if os.path.isdir(models_dir):
for subdir in os.listdir(models_dir):
if os.path.isdir(os.path.join(models_dir, subdir)):
models.append(subdir)
return gr.update(value=model if model in models else None, choices=models, visible=True)
refresh_models.click(refresh_all_models, model, model)