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

Update A1111MetadataParser to correctly extract model ID, name, and version from Civitai info

Browse Source 
- Changed the extraction of model ID to use 'id' instead of 'modelVersionId'.
- Updated the retrieval of model name and version to align with the new Civitai response structure, ensuring accurate metadata parsing for LoRA entries.
- Improved error handling and logging for better traceability during metadata fetching.
This commit is contained in:
Will Miao
2025-03-19 05:49:53 +08:00
parent 541ad624c5
commit 0017a6cce5
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
py/utils/recipe_parsers.py
View File

@@ -430,11 +430,11 @@ class A1111MetadataParser(RecipeMetadataParser):
civitai_info = await civitai_client.get_model_by_hash(hash_value)
if civitai_info and civitai_info.get("error") != "Model not found":
# Get model version ID
lora_entry['id'] = civitai_info.get('modelVersionId', '')
lora_entry['id'] = civitai_info.get('id', '')
# Get model name and version
lora_entry['name'] = civitai_info.get('modelName', lora_name)
lora_entry['version'] = civitai_info.get('modelVersionName', '')
lora_entry['name'] = civitai_info.get('model', {}).get('name', lora_name)
lora_entry['version'] = civitai_info.get('name', '')
# Get thumbnail URL
if 'images' in civitai_info and civitai_info['images']:
@@ -476,7 +476,7 @@ class A1111MetadataParser(RecipeMetadataParser):
logger.error(f"Error fetching Civitai info for LoRA hash {hash_value}: {e}")
loras.append(lora_entry)
# Set base_model to the most common one from civitai_info
base_model = None
if base_model_counts: