File tree Expand file tree Collapse file tree 1 file changed +8
-2
lines changed Expand file tree Collapse file tree 1 file changed +8
-2
lines changed Original file line number Diff line number Diff line change @@ -136,9 +136,15 @@ def print_model_info():
136
136
# Import here to avoid circular imports
137
137
try :
138
138
from ..config import get_env_var
139
+ from ..model_manager import ModelManager
139
140
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"
142
148
143
149
# Get optimization settings
144
150
enable_quantization = get_env_var ("LOCALLAB_ENABLE_QUANTIZATION" , default = "false" ).lower () == "true"
You can’t perform that action at this time.
0 commit comments