oliver-aizip commited on
Commit
dc2d37a
·
1 Parent(s): c0f44db

add additional models

Browse files
Files changed (1) hide show
  1. utils/models.py +4 -1
utils/models.py CHANGED
@@ -9,13 +9,16 @@ from .shared import generation_interrupt
9
 
10
  models = {
11
  "Qwen2.5-1.5b-Instruct": "qwen/qwen2.5-1.5b-instruct",
12
- "Qwen2.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
 
9
 
10
  models = {
11
  "Qwen2.5-1.5b-Instruct": "qwen/qwen2.5-1.5b-instruct",
12
+ "Qwen2.5-3b-Instruct": "qwen/qwen2.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
+ "Cogito-v1-preview-llama-3b": "deepcogito/cogito-v1-preview-llama-3b",
20
+ "IBM Granite-3.3-2b-instruct": "ibm-granite/granite-3.3-2b-instruct",
21
+
22
  }
23
 
24
  # List of model names for easy access