ABVM commited on
Commit
a0cc714
·
verified ·
1 Parent(s): f50775f

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +3 -3
agent.py CHANGED
@@ -8,10 +8,10 @@ class GaiaAgent:
8
  token = os.getenv("hf_token")
9
  print("DEBUG: hf_token is", token[:6]+ "...")
10
  self.model = InferenceClientModel(
11
- max_tokens = 2048,
12
  temperature = 0.5,
13
- model_id = "meta-llama/Llama-3.3-70B-Instruct",
14
- custom_role_conversions=None,
15
  provider="hf-inference"
16
  )
17
 
 
8
  token = os.getenv("hf_token")
9
  print("DEBUG: hf_token is", token[:6]+ "...")
10
  self.model = InferenceClientModel(
11
+ max_tokens = 1024,
12
  temperature = 0.5,
13
+ model_id = "meta-llama/Llama-2-7b-chat-hf",
14
+ #custom_role_conversions=None,
15
  provider="hf-inference"
16
  )
17