diff --git a/modules/ui_model_menu.py b/modules/ui_model_menu.py index f87b680a..34d58177 100644 --- a/modules/ui_model_menu.py +++ b/modules/ui_model_menu.py @@ -260,6 +260,8 @@ def download_model_wrapper(repo_id, specific_file, progress=gr.Progress(), retur yield ("Please enter a model path") return + repo_id = repo_id.strip() + specific_file = specific_file.strip() downloader = importlib.import_module("download-model").ModelDownloader() progress(0.0)