Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
cb092d7
1
Parent(s):
db5316a
remove gated models for now
Browse files- utils/models.py +2 -1
utils/models.py
CHANGED
@@ -13,7 +13,8 @@ from .prompts import format_rag_prompt
|
|
13 |
|
14 |
models = {
|
15 |
"Qwen2.5-1.5b-Instruct": "qwen/qwen2.5-1.5b-instruct",
|
16 |
-
"
|
|
|
17 |
#TODO add more models
|
18 |
}
|
19 |
|
|
|
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 |
}
|
20 |
|