another fix to repo_id loading

This commit is contained in:
Lincoln Stein 2023-07-02 09:18:11 -04:00
parent b2775d6b4c
commit 2d314d2b3d

View File

@ -206,7 +206,7 @@ class ModelInstall(object):
# huggingface repo # huggingface repo
elif len(str(model_path_id_or_url).split('/')) == 2: elif len(str(model_path_id_or_url).split('/')) == 2:
models_installed.add(self._install_repo(str(path))) models_installed.add(self._install_repo(str(model_path_id_or_url)))
# a URL # a URL
elif model_path_id_or_url.startswith(("http:", "https:", "ftp:")): elif model_path_id_or_url.startswith(("http:", "https:", "ftp:")):