Skip to content

Commit 51a56df

Browse files
committed
Case insensitive priority model check
1 parent d8f59a7 commit 51a56df

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

interpreter/terminal_interface/local_setup.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -252,9 +252,9 @@ def download_model(models_dir, models, interpreter):
252252
priority_models=["llama3","codestral"]
253253
priority_models_found=[]
254254
for word in priority_models:
255-
models_to_move=[name for name in names if word in name]
255+
models_to_move=[name for name in names if word.lower() in name.lower()]
256256
priority_models_found.extend(models_to_move)
257-
names=[name for name in names if not any(word in name for word in priority_models)]
257+
names=[name for name in names if not any(word.lower() in name.lower() for word in priority_models)]
258258
names=priority_models_found+names
259259

260260
for model in ["llama3", "phi3", "wizardlm2", "codestral"]:

0 commit comments

Comments
 (0)