tfrere commited on
Commit
ab8d3f5
·
1 Parent(s): 81e0b0c

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": ["deepseek-ai/DeepSeek-R1-Distill-Llama-70B"],
171
- "summarization": ["deepseek-ai/DeepSeek-R1-Distill-Llama-70B"],
172
  "chunking": ["intfloat/multilingual-e5-large-instruct"],
173
- "single_shot_question_generation": ["deepseek-ai/DeepSeek-R1-Distill-Llama-70B"],
174
- "multi_hop_question_generation": ["deepseek-ai/DeepSeek-R1-Distill-Llama-70B"],
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": {