git/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2026-04-26 16:52:08 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix error message when adding a local path with quotes around the string

Browse Source
This commit is contained in:
David Hauptman 2024-12-03 15:34:49 -06:00 committed by psychedelicious
parent 174ea021a6
commit 6fc6be3aa0
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
invokeai/app/services/model_install/model_install_default.py
View File

@ -438,9 +438,10 @@ class ModelInstallService(ModelInstallServiceBase):
variants = "|".join(ModelRepoVariant.__members__.values())
hf_repoid_re = f"^([^/:]+/[^/:]+)(?::({variants})?(?::/?([^:]+))?)?$"
source_obj: Optional[StringLikeSource] = None
source_stripped = source.strip('\"') # Strip possible quotes from source string to avoid later errors with local files or directories
if Path(source).exists(): # A local file or directory
source_obj = LocalModelSource(path=Path(source))
if Path(source_stripped).exists(): # A local file or directory
source_obj = LocalModelSource(path=Path(source_stripped))
elif match := re.match(hf_repoid_re, source):
source_obj = HFModelSource(
repo_id=match.group(1),