Skip to content

Commit 1c7ae13

Browse files
committed
fixed banner incorrect model name
1 parent f4c4079 commit 1c7ae13

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

locallab/ui/banners.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -136,9 +136,15 @@ def print_model_info():
136136
# Import here to avoid circular imports
137137
try:
138138
from ..config import get_env_var
139+
from ..model_manager import ModelManager
139140

140-
# Get model information
141-
model_id = get_env_var("HUGGINGFACE_MODEL", default="microsoft/phi-2")
141+
# Get model information from model manager first
142+
model_manager = ModelManager()
143+
model_id = model_manager.current_model if model_manager.current_model else None
144+
145+
# If no model loaded, check environment/config
146+
if not model_id:
147+
model_id = get_env_var("HUGGINGFACE_MODEL") or get_env_var("LOCALLAB_MODEL_ID") or "microsoft/phi-2"
142148

143149
# Get optimization settings
144150
enable_quantization = get_env_var("LOCALLAB_ENABLE_QUANTIZATION", default="false").lower() == "true"

0 commit comments

Comments
 (0)