diff --git a/scripts/t2v_helpers/args.py b/scripts/t2v_helpers/args.py index 3496382..66dbf0d 100644 --- a/scripts/t2v_helpers/args.py +++ b/scripts/t2v_helpers/args.py @@ -87,12 +87,13 @@ def setup_text2video_settings_dictionary(): models = [] if os.path.isdir(os.path.join(ph.models_path, 'ModelScope/t2v')): models.append('') - if os.path.isdir(os.path.join(ph.models_path, 'VideoCrafter')): + if os.path.isdir(os.path.join(ph.models_path, 'VideoCrafter/')): models.append('') 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)