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

Fixed an issue caused by inconsistent base model name for Illustrious. It fixes https://github.com/willmiao/ComfyUI-Lora-Manager/issues/37

Browse Source
This commit is contained in:
Will Miao
2025-03-13 11:00:55 +08:00
parent c88388da67
commit e57a72d12b
4 changed files with 16 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
py/routes/api_routes.py
View File

@@ -4,6 +4,8 @@ import logging
from aiohttp import web from aiohttp import web
from typing import Dict, List from typing import Dict, List
from ..utils.model_utils import determine_base_model
from ..services.file_monitor import LoraFileMonitor from ..services.file_monitor import LoraFileMonitor
from ..services.download_manager import DownloadManager from ..services.download_manager import DownloadManager
from ..services.civitai_client import CivitaiClient from ..services.civitai_client import CivitaiClient
@@ -351,8 +353,8 @@ class ApiRoutes:
# Update model name if available # Update model name if available
if 'model' in civitai_metadata: if 'model' in civitai_metadata:
local_metadata['model_name'] = civitai_metadata['model'].get('name', if civitai_metadata.get('model', {}).get('name'):
local_metadata.get('model_name')) local_metadata['model_name'] = determine_base_model(civitai_metadata['model']['name'])
# Fetch additional model metadata (description and tags) if we have model ID # Fetch additional model metadata (description and tags) if we have model ID
model_id = civitai_metadata['modelId'] model_id = civitai_metadata['modelId']

8
py/utils/file_utils.py
View File

@@ -4,6 +4,8 @@ import hashlib
import json import json
from typing import Dict, Optional from typing import Dict, Optional
from .model_utils import determine_base_model
from .lora_metadata import extract_lora_metadata from .lora_metadata import extract_lora_metadata
from .models import LoraMetadata from .models import LoraMetadata
@@ -105,6 +107,12 @@ async def load_metadata(file_path: str) -> Optional[LoraMetadata]:
data = json.load(f) data = json.load(f)
needs_update = False needs_update = False
# Check and normalize base model name
normalized_base_model = determine_base_model(data['base_model'])
if data['base_model'] != normalized_base_model:
data['base_model'] = normalized_base_model
needs_update = True
# Compare paths without extensions # Compare paths without extensions
stored_path_base = os.path.splitext(data['file_path'])[0] stored_path_base = os.path.splitext(data['file_path'])[0]

3
py/utils/model_utils.py
View File

@@ -8,7 +8,8 @@ BASE_MODEL_MAPPING = {
"sd-v2": "SD 2.0", "sd-v2": "SD 2.0",
"flux1": "Flux.1 D", "flux1": "Flux.1 D",
"flux.1 d": "Flux.1 D", "flux.1 d": "Flux.1 D",
"illustrious": "IL", "illustrious": "Illustrious",
"il": "Illustrious",
"pony": "Pony", "pony": "Pony",
"Hunyuan Video": "Hunyuan Video" "Hunyuan Video": "Hunyuan Video"
} }

4
static/js/utils/constants.js
View File

@@ -31,7 +31,7 @@ export const BASE_MODELS = {
LUMINA: "Lumina", LUMINA: "Lumina",
KOLORS: "Kolors", KOLORS: "Kolors",
NOOBAI: "NoobAI", NOOBAI: "NoobAI",
IL: "IL", ILLUSTRIOUS: "Illustrious",
PONY: "Pony", PONY: "Pony",
// Video models // Video models
@@ -82,7 +82,7 @@ export const BASE_MODEL_CLASSES = {
[BASE_MODELS.LUMINA]: "lumina", [BASE_MODELS.LUMINA]: "lumina",
[BASE_MODELS.KOLORS]: "kolors", [BASE_MODELS.KOLORS]: "kolors",
[BASE_MODELS.NOOBAI]: "noobai", [BASE_MODELS.NOOBAI]: "noobai",
[BASE_MODELS.IL]: "il", [BASE_MODELS.ILLUSTRIOUS]: "il",
[BASE_MODELS.PONY]: "pony", [BASE_MODELS.PONY]: "pony",
// Default // Default