alphard/ComfyUI-Lora-Manager
1
0
Fork 0
mirror of https://github.com/willmiao/ComfyUI-Lora-Manager.git synced 2026-03-21 21:22:11 -03:00
Code Issues Packages Projects Releases Wiki Activity

fix(download-manager): improve handling of civitai payloads to avoid empty dictionaries

Browse Source
This commit is contained in:
Will Miao
2025-10-05 07:29:11 +08:00
parent 9d22dd3465
commit 98425f37b8
1 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
py/utils/example_images_download_manager.py
View File

@@ -358,7 +358,8 @@ class DownloadManager:
full_model = await MetadataUpdater.get_updated_model(
model_hash, scanner
)
civitai_payload = (full_model or {}).get('civitai', {}) if full_model else {}
civitai_payload = (full_model or {}).get('civitai') if full_model else None
civitai_payload = civitai_payload or {}
# If no local images, try to download from remote
if civitai_payload.get('images'):
@@ -378,7 +379,8 @@ class DownloadManager:
updated_model = await MetadataUpdater.get_updated_model(
model_hash, scanner
)
updated_civitai = (updated_model or {}).get('civitai', {}) if updated_model else {}
updated_civitai = (updated_model or {}).get('civitai') if updated_model else None
updated_civitai = updated_civitai or {}
if updated_civitai.get('images'):
# Retry download with updated metadata
@@ -647,7 +649,8 @@ class DownloadManager:
full_model = await MetadataUpdater.get_updated_model(
model_hash, scanner
)
civitai_payload = (full_model or {}).get('civitai', {}) if full_model else {}
civitai_payload = (full_model or {}).get('civitai') if full_model else None
civitai_payload = civitai_payload or {}
# If no local images, try to download from remote
if civitai_payload.get('images'):
@@ -667,7 +670,8 @@ class DownloadManager:
updated_model = await MetadataUpdater.get_updated_model(
model_hash, scanner
)
updated_civitai = (updated_model or {}).get('civitai', {}) if updated_model else {}
updated_civitai = (updated_model or {}).get('civitai') if updated_model else None
updated_civitai = updated_civitai or {}
if updated_civitai.get('images'):
# Retry download with updated metadata