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

feat(model-scanner): enhance page type determination for model types

Browse Source
This commit is contained in:
Will Miao
2025-10-04 17:08:02 +08:00
parent 0aa6c53c1f
commit 056f727bfd
2 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
py/routes/handlers/misc_handlers.py
View File

@@ -22,7 +22,6 @@ from aiohttp import web
from ...config import config from ...config import config
from ...services.metadata_service import ( from ...services.metadata_service import (
get_metadata_archive_manager, get_metadata_archive_manager,
get_metadata_provider,
update_metadata_providers, update_metadata_providers,
) )
from ...services.service_registry import ServiceRegistry from ...services.service_registry import ServiceRegistry

7
py/services/model_scanner.py
View File

@@ -211,7 +211,12 @@ class ModelScanner:
self._is_initializing = True self._is_initializing = True
# Determine the page type based on model type # Determine the page type based on model type
page_type = 'loras' if self.model_type == 'lora' else 'checkpoints' page_type_map = {
'lora': 'loras',
'checkpoint': 'checkpoints',
'embedding': 'embeddings'
}
page_type = page_type_map.get(self.model_type, self.model_type)
# First, try to load from cache # First, try to load from cache
await ws_manager.broadcast_init_progress({ await ws_manager.broadcast_init_progress({