Spaces:
Sleeping
Sleeping
EtienneB
commited on
Commit
·
d55b1b4
1
Parent(s):
263c392
Update agent.py
Browse files
agent.py
CHANGED
|
@@ -143,7 +143,9 @@ def build_graph():
|
|
| 143 |
"""Build the graph"""
|
| 144 |
# First create the HuggingFaceEndpoint
|
| 145 |
llm_endpoint = HuggingFaceEndpoint(
|
| 146 |
-
repo_id="Qwen/Qwen2.5-14B-Instruct",
|
|
|
|
|
|
|
| 147 |
huggingfacehub_api_token=HUGGINGFACEHUB_API_TOKEN,
|
| 148 |
temperature=0.1, # Maximum determinism
|
| 149 |
max_new_tokens=128, # Even more restrictive
|
|
|
|
| 143 |
"""Build the 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",
|
| 149 |
huggingfacehub_api_token=HUGGINGFACEHUB_API_TOKEN,
|
| 150 |
temperature=0.1, # Maximum determinism
|
| 151 |
max_new_tokens=128, # Even more restrictive
|