lwant commited on
Commit
e777bbb
Β·
1 Parent(s): a32f635

Add `light_model_name` option for extended model configuration in `agent.py`

Browse files
Files changed (1) hide show
  1. src/gaia_solving_agent/agent.py +1 -0
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