Spaces:
Sleeping
Sleeping
EtienneB
commited on
Commit
·
ba0e6f8
1
Parent(s):
d49a17e
Update agent.py
Browse files
agent.py
CHANGED
|
@@ -144,8 +144,9 @@ def build_graph():
|
|
| 144 |
# First create the HuggingFaceEndpoint
|
| 145 |
llm_endpoint = HuggingFaceEndpoint(
|
| 146 |
# repo_id="Qwen/Qwen2.5-14B-Instruct",
|
| 147 |
-
repo_id="HuggingFaceH4/mistral-7b-anthropic",
|
| 148 |
# repo_id="Qwen/Qwen2.5-Coder-32B-Instruct", # this one is poor
|
|
|
|
| 149 |
huggingfacehub_api_token=HUGGINGFACEHUB_API_TOKEN,
|
| 150 |
temperature=0.1, # Maximum determinism
|
| 151 |
max_new_tokens=512, # Even more restrictive with 128
|
|
|
|
| 144 |
# First create the HuggingFaceEndpoint
|
| 145 |
llm_endpoint = HuggingFaceEndpoint(
|
| 146 |
# repo_id="Qwen/Qwen2.5-14B-Instruct",
|
| 147 |
+
# repo_id="HuggingFaceH4/mistral-7b-anthropic", # wrong id?
|
| 148 |
# repo_id="Qwen/Qwen2.5-Coder-32B-Instruct", # this one is poor
|
| 149 |
+
repo_id="meta-llama/Llama-3.1-8B",
|
| 150 |
huggingfacehub_api_token=HUGGINGFACEHUB_API_TOKEN,
|
| 151 |
temperature=0.1, # Maximum determinism
|
| 152 |
max_new_tokens=512, # Even more restrictive with 128
|