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

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

View File

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

View File

@@ -211,7 +211,12 @@ class ModelScanner:
self._is_initializing = True
# 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
await ws_manager.broadcast_init_progress({