Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Commit
·
c2cd706
1
Parent(s):
ec6e669
Fix model logging to use display_name instead of name field
Browse files- db/models.py +1 -1
- models/loader.py +1 -1
db/models.py
CHANGED
@@ -78,7 +78,7 @@ async def fetch_model_by_id(model_id: str) -> Union[Dict[str, Any], None]:
|
|
78 |
row = await conn.fetchrow(query, model_id)
|
79 |
|
80 |
if row:
|
81 |
-
logger.info(f"Found model with ID '{model_id}': {row['
|
82 |
return dict(row)
|
83 |
else:
|
84 |
logger.warning(f"No model found with ID '{model_id}'")
|
|
|
78 |
row = await conn.fetchrow(query, model_id)
|
79 |
|
80 |
if row:
|
81 |
+
logger.info(f"Found model with ID '{model_id}': {row['display_name']}")
|
82 |
return dict(row)
|
83 |
else:
|
84 |
logger.warning(f"No model found with ID '{model_id}'")
|
models/loader.py
CHANGED
@@ -106,7 +106,7 @@ async def load_models(models_data: List[Dict[str, Any]]) -> None:
|
|
106 |
except Exception as e:
|
107 |
# Log l'échec mais continue avec les autres modèles
|
108 |
logger.error(f"Failed to load model ID {model_id}: {e}")
|
109 |
-
failed_models.append(model_data.get('
|
110 |
|
111 |
logger.info(f"Finished loading models. Successfully loaded: {loaded_count}/{len(models_data)}")
|
112 |
if failed_models:
|
|
|
106 |
except Exception as e:
|
107 |
# Log l'échec mais continue avec les autres modèles
|
108 |
logger.error(f"Failed to load model ID {model_id}: {e}")
|
109 |
+
failed_models.append(model_data.get('display_name', f'ID {model_id}'))
|
110 |
|
111 |
logger.info(f"Finished loading models. Successfully loaded: {loaded_count}/{len(models_data)}")
|
112 |
if failed_models:
|