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

feat: Enhance model file retrieval by ensuring primary model is selected from files list. Fixes https://github.com/willmiao/ComfyUI-Lora-Manager/issues/127

Browse Source
This commit is contained in:
Will Miao
2025-04-25 05:45:29 +08:00
parent 23c9a98f66
commit 0a52575e8b
2 changed files with 9 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
py/services/download_manager.py
View File

@@ -88,16 +88,16 @@ class DownloadManager:
version_info = None version_info = None
error_msg = None error_msg = None
if download_url: if model_hash:
# Extract version ID from download URL # Get model by hash
version_id = download_url.split('/')[-1] version_info = await civitai_client.get_model_by_hash(model_hash)
version_info, error_msg = await civitai_client.get_model_version_info(version_id)
elif model_version_id: elif model_version_id:
# Use model version ID directly # Use model version ID directly
version_info, error_msg = await civitai_client.get_model_version_info(model_version_id) version_info, error_msg = await civitai_client.get_model_version_info(model_version_id)
elif model_hash: elif download_url:
# Get model by hash # Extract version ID from download URL
version_info = await civitai_client.get_model_by_hash(model_hash) version_id = download_url.split('/')[-1]
version_info, error_msg = await civitai_client.get_model_version_info(version_id)
if not version_info: if not version_info:

3
py/utils/recipe_parsers.py
View File

@@ -97,8 +97,9 @@ class RecipeMetadataParser(ABC):
# Process file information if available # Process file information if available
if 'files' in civitai_info: if 'files' in civitai_info:
# Find the primary model file (type="Model" and primary=true) in the files list
model_file = next((file for file in civitai_info.get('files', []) model_file = next((file for file in civitai_info.get('files', [])
if file.get('type') == 'Model'), None) if file.get('type') == 'Model' and file.get('primary') == True), None)
if model_file: if model_file:
# Get size # Get size