ABVM commited on
Commit
f7d0260
·
verified ·
1 Parent(s): 280b18f

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -1
agent.py CHANGED
@@ -5,7 +5,7 @@ from smolagents import VisitWebpageTool, WebSearchTool, WikipediaSearchTool, Pyt
5
  class GaiaAgent:
6
  def __init__(self):
7
  self.model = InferenceClientModel(
8
- max_tokens = 8096,
9
  temperature = 0.5,
10
  model_id = "Qwen/Qwen2.5-Coder-32B-Instruct",
11
  provider = "huggingface", #'https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud',
 
5
  class GaiaAgent:
6
  def __init__(self):
7
  self.model = InferenceClientModel(
8
+ max_tokens = 2048,
9
  temperature = 0.5,
10
  model_id = "Qwen/Qwen2.5-Coder-32B-Instruct",
11
  provider = "huggingface", #'https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud',