Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
update models
Browse files
backend/tasks/create_bench_config_file.py
CHANGED
@@ -124,7 +124,7 @@ class CreateBenchConfigTask:
|
|
124 |
# "Qwen/Qwen2.5-72B-Instruct"
|
125 |
# "meta-llama/Llama-3.1-8B-Instruct"
|
126 |
# "Qwen/Qwen2.5-32B-Instruct",
|
127 |
-
"deepseek-ai/DeepSeek-R1-Distill-Llama-70B",
|
128 |
]
|
129 |
|
130 |
# Track found models
|
@@ -167,11 +167,11 @@ class CreateBenchConfigTask:
|
|
167 |
"model_list": model_list,
|
168 |
|
169 |
"model_roles": {
|
170 |
-
"ingestion": ["
|
171 |
-
"summarization": ["
|
172 |
"chunking": ["intfloat/multilingual-e5-large-instruct"],
|
173 |
-
"single_shot_question_generation": ["
|
174 |
-
"multi_hop_question_generation": ["
|
175 |
},
|
176 |
"pipeline": {
|
177 |
"ingestion": {
|
|
|
124 |
# "Qwen/Qwen2.5-72B-Instruct"
|
125 |
# "meta-llama/Llama-3.1-8B-Instruct"
|
126 |
# "Qwen/Qwen2.5-32B-Instruct",
|
127 |
+
# "deepseek-ai/DeepSeek-R1-Distill-Llama-70B",
|
128 |
]
|
129 |
|
130 |
# Track found models
|
|
|
167 |
"model_list": model_list,
|
168 |
|
169 |
"model_roles": {
|
170 |
+
"ingestion": ["Qwen/Qwen2.5-32B-Instruct"],
|
171 |
+
"summarization": ["Qwen/Qwen2.5-32B-Instruct"],
|
172 |
"chunking": ["intfloat/multilingual-e5-large-instruct"],
|
173 |
+
"single_shot_question_generation": ["Qwen/Qwen2.5-32B-Instruct"],
|
174 |
+
"multi_hop_question_generation": ["Qwen/Qwen2.5-32B-Instruct"],
|
175 |
},
|
176 |
"pipeline": {
|
177 |
"ingestion": {
|