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": {
|