oliver-aizip commited on
Commit
dab8aab
·
1 Parent(s): cb092d7

fix models

Browse files
Files changed (1) hide show
  1. utils/models.py +1 -1
utils/models.py CHANGED
@@ -13,7 +13,7 @@ from .prompts import format_rag_prompt
13
 
14
  models = {
15
  "Qwen2.5-1.5b-Instruct": "qwen/qwen2.5-1.5b-instruct",
16
- "Qwen2.5-3b-Instruct": "qwen/qwen2.5-1.5b-instruct", # remove gated for now
17
  #"Llama-3.2-1b-Instruct": "meta-llama/llama-3.2-1b-instruct",
18
  #TODO add more models
19
  }
 
13
 
14
  models = {
15
  "Qwen2.5-1.5b-Instruct": "qwen/qwen2.5-1.5b-instruct",
16
+ "Qwen2.5-3b-Instruct": "qwen/qwen2.5-3b-instruct", # remove gated for now
17
  #"Llama-3.2-1b-Instruct": "meta-llama/llama-3.2-1b-instruct",
18
  #TODO add more models
19
  }