cpatino10 commited on
Commit
cc15036
·
verified ·
1 Parent(s): ea9859d

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
agent.py CHANGED
@@ -1,4 +1,4 @@
1
- from smolagents import CodeAgent, InferenceClientModel, HfApiModel
2
  from tools import all_tools
3
 
4
  # define GAIA specific system prompt
@@ -12,7 +12,7 @@ If you are asked for a comma separated list, apply the above rules depending of
12
  Return ONLY the final answer as the output of your last step."""
13
 
14
  # model
15
- model = HfApiModel(
16
  model_id="Qwen/Qwen2.5-Coder-1.5B-Instruct"
17
  )
18
 
 
1
+ from smolagents import CodeAgent, InferenceClientModel
2
  from tools import all_tools
3
 
4
  # define GAIA specific system prompt
 
12
  Return ONLY the final answer as the output of your last step."""
13
 
14
  # model
15
+ model = InferenceClientModel(
16
  model_id="Qwen/Qwen2.5-Coder-1.5B-Instruct"
17
  )
18