igor04091968 commited on
Commit
41124ca
·
1 Parent(s): 6cb6ed0

Fix: Use default model name if env var is empty

Browse files
Files changed (1) hide show
  1. utils.py +2 -2
utils.py CHANGED
@@ -15,8 +15,8 @@ models_config = {
15
  "Apriel-1.5-15B-thinker": {
16
  "MODEL_DISPLAY_NAME": "Apriel-1.5-15B-thinker",
17
  "MODEL_HF_URL": "https://huggingface.co/ServiceNow-AI/Apriel-1.5-15b-Thinker",
18
- "MODEL_NAME": os.environ.get("MODEL_NAME_APRIEL_1_5_15B", "Apriel-1.5-15B-thinker"),
19
- "VLLM_API_URL": os.environ.get("VLLM_API_URL_APRIEL_1_5_15B", "http://localhost:8080/v1"),
20
  "VLLM_API_URL_LIST": os.environ.get("VLLM_API_URL_LIST_APRIEL_1_5_15B"),
21
  "REASONING": True,
22
  "MULTIMODAL": True,
 
15
  "Apriel-1.5-15B-thinker": {
16
  "MODEL_DISPLAY_NAME": "Apriel-1.5-15B-thinker",
17
  "MODEL_HF_URL": "https://huggingface.co/ServiceNow-AI/Apriel-1.5-15b-Thinker",
18
+ "MODEL_NAME": os.environ.get("MODEL_NAME_APRIEL_1_5_15B") or "Apriel-1.5-15B-thinker",
19
+ "VLLM_API_URL": os.environ.get("VLLM_API_URL_APRIEL_1_5_15B") or "http://localhost:8080/v1",
20
  "VLLM_API_URL_LIST": os.environ.get("VLLM_API_URL_LIST_APRIEL_1_5_15B"),
21
  "REASONING": True,
22
  "MULTIMODAL": True,