ABVM commited on
Commit
aad1e32
·
verified ·
1 Parent(s): 3e7fa24

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -1
agent.py CHANGED
@@ -12,7 +12,7 @@ class GaiaAgent:
12
  self.model = InferenceClientModel(
13
  max_tokens = 1024,
14
  temperature = 0.5,
15
- model_id = 'https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud', #"meta-llama/Llama-3.1-8B-Instruct",
16
  #custom_role_conversions=None,
17
  provider="hf-inference",
18
  token=token
 
12
  self.model = InferenceClientModel(
13
  max_tokens = 1024,
14
  temperature = 0.5,
15
+ model_id = 'meta-llama/Llama-3.1-8B-Instruct',
16
  #custom_role_conversions=None,
17
  provider="hf-inference",
18
  token=token