diff --git a/py/routes/handlers/misc_handlers.py b/py/routes/handlers/misc_handlers.py index caaf3c66..398013dc 100644 --- a/py/routes/handlers/misc_handlers.py +++ b/py/routes/handlers/misc_handlers.py @@ -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 diff --git a/py/services/model_scanner.py b/py/services/model_scanner.py index 61917736..4e1c0eac 100644 --- a/py/services/model_scanner.py +++ b/py/services/model_scanner.py @@ -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({