Spaces:
Sleeping
Sleeping
Update agent.py
Browse files
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 =
|
| 12 |
temperature = 0.5,
|
| 13 |
-
model_id = "meta-llama/Llama-
|
| 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 |
|