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

feat: Add Civitai model tags for prioritized subfolder organization in download manager

Browse Source
This commit is contained in:
Will Miao
2025-07-12 10:32:15 +08:00
parent 0dbb76e8c8
commit 7e0df4d718
2 changed files with 24 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
py/services/download_manager.py
View File

@@ -3,7 +3,7 @@ import os
import asyncio import asyncio
from typing import Dict from typing import Dict
from ..utils.models import LoraMetadata, CheckpointMetadata from ..utils.models import LoraMetadata, CheckpointMetadata
from ..utils.constants import CARD_PREVIEW_WIDTH, VALID_LORA_TYPES from ..utils.constants import CARD_PREVIEW_WIDTH, VALID_LORA_TYPES, CIVITAI_MODEL_TAGS
from ..utils.exif_utils import ExifUtils from ..utils.exif_utils import ExifUtils
from ..utils.metadata_manager import MetadataManager from ..utils.metadata_manager import MetadataManager
from .service_registry import ServiceRegistry from .service_registry import ServiceRegistry
@@ -126,12 +126,24 @@ class DownloadManager:
return {'success': False, 'error': 'Default lora root path not set in settings'} return {'success': False, 'error': 'Default lora root path not set in settings'}
save_dir = default_path save_dir = default_path
# Set relative_path to version_info.baseModel/first_tag if available # Set relative_path to version_info.baseModel/prioritized_tag
base_model = version_info.get('baseModel', '') base_model = version_info.get('baseModel', '')
model_tags = version_info.get('model', {}).get('tags', []) model_tags = version_info.get('model', {}).get('tags', [])
if base_model: if base_model:
if model_tags: # Find the first Civitai model tag that exists in model_tags
relative_path = os.path.join(base_model, model_tags[0]) prioritized_tag = None
for civitai_tag in CIVITAI_MODEL_TAGS:
if civitai_tag in model_tags:
prioritized_tag = civitai_tag
break
# If no Civitai model tag found, fallback to first tag
if prioritized_tag is None and model_tags:
prioritized_tag = model_tags[0]
if prioritized_tag:
relative_path = os.path.join(base_model, prioritized_tag)
else: else:
relative_path = base_model relative_path = base_model

9
py/utils/constants.py
View File

@@ -46,4 +46,11 @@ SUPPORTED_MEDIA_EXTENSIONS = {
} }
# Valid Lora types # Valid Lora types
VALID_LORA_TYPES = ['lora', 'locon', 'dora'] VALID_LORA_TYPES = ['lora', 'locon', 'dora']
# Civitai model tags in priority order for subfolder organization
CIVITAI_MODEL_TAGS = [
'character', 'style', 'concept', 'clothing', 'base model',
'poses', 'background', 'tool', 'vehicle', 'buildings',
'objects', 'assets', 'animal', 'action'
]