Spaces:
Runtime error
Runtime error
Add `light_model_name` option for extended model configuration in `agent.py`
Browse files
src/gaia_solving_agent/agent.py
CHANGED
|
@@ -22,6 +22,7 @@ from gaia_solving_agent.utils import extract_pattern
|
|
| 22 |
|
| 23 |
# Choice of the model
|
| 24 |
cheap_model_name = "meta-llama/Meta-Llama-3.1-8B-Instruct"
|
|
|
|
| 25 |
balanced_model_name = "meta-llama/Meta-Llama-3.1-70B-Instruct"
|
| 26 |
reasoning_model_name = "deepseek-ai/DeepSeek-R1-0528"
|
| 27 |
vlm_model_name = "mistralai/Mistral-Small-3.1-24B-Instruct-2503" # For VLM needs
|
|
|
|
| 22 |
|
| 23 |
# Choice of the model
|
| 24 |
cheap_model_name = "meta-llama/Meta-Llama-3.1-8B-Instruct"
|
| 25 |
+
light_model_name = "Qwen/Qwen2.5-32B-Instruct"
|
| 26 |
balanced_model_name = "meta-llama/Meta-Llama-3.1-70B-Instruct"
|
| 27 |
reasoning_model_name = "deepseek-ai/DeepSeek-R1-0528"
|
| 28 |
vlm_model_name = "mistralai/Mistral-Small-3.1-24B-Instruct-2503" # For VLM needs
|