oliver-aizip commited on
Commit
533ae49
·
1 Parent(s): 97629be

update model list

Browse files
Files changed (1) hide show
  1. utils/models.py +6 -1
utils/models.py CHANGED
@@ -9,8 +9,13 @@ from .shared import generation_interrupt
9
 
10
  models = {
11
  "Qwen2.5-1.5b-Instruct": "qwen/qwen2.5-1.5b-instruct",
12
- "Llama-3.2-1b-Instruct": "meta-llama/llama-3.2-1b-instruct",
 
 
13
  "Gemma-3-1b-it": "google/gemma-3-1b-it",
 
 
 
14
  }
15
 
16
  # List of model names for easy access
 
9
 
10
  models = {
11
  "Qwen2.5-1.5b-Instruct": "qwen/qwen2.5-1.5b-instruct",
12
+ "Qwen-2.5-3b-Instruct": "qwen/qwen-2.5-3b-instruct",
13
+ #"Llama-3.2-1b-Instruct": "meta-llama/llama-3.2-1b-instruct",
14
+ #"Llama-3.2-3b-Instruct": "meta-llama/llama-3.2-3b-instruct",
15
  "Gemma-3-1b-it": "google/gemma-3-1b-it",
16
+ "Gemma-3-4b-it": "google/gemma-3-4b-it",
17
+ "Gemma-2-2b-it": "google/gemma-2-2b-it",
18
+ "Phi-4-mini-instruct": "microsoft/phi-4-mini-instruct",
19
  }
20
 
21
  # List of model names for easy access