Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix ruff styling

Browse Source
This commit is contained in:
Brandon Rising 2024-01-30 22:32:40 -05:00 committed by Kent Keirsey
parent 522ff4a042
commit a0996b1c0a
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
invokeai/backend/install/model_install_backend.py
View File

@ -330,7 +330,9 @@ class ModelInstall(object):
with TemporaryDirectory(dir=self.config.models_path) as staging:
CIVITAI_RE = r".*civitai.com.*"
civit_url = re.match(CIVITAI_RE, url, re.IGNORECASE)
location = download_with_resume(url, Path(staging), access_token=self.civitai_api_key if civit_url else None)
location = download_with_resume(
url, Path(staging), access_token=self.civitai_api_key if civit_url else None
)
if not location:
logger.error(f"Unable to download {url}. Skipping.")
info = ModelProbe().heuristic_probe(location, self.prediction_helper)