Spaces:
Sleeping
Sleeping
EtienneB
commited on
Commit
·
7faf23e
1
Parent(s):
c4e5a43
Update agent.py
Browse files
agent.py
CHANGED
@@ -99,11 +99,11 @@ def build_graph():
|
|
99 |
"""Build the graph"""
|
100 |
# First create the HuggingFaceEndpoint
|
101 |
llm_endpoint = HuggingFaceEndpoint(
|
102 |
-
repo_id="
|
103 |
huggingfacehub_api_token=HUGGINGFACEHUB_API_TOKEN,
|
104 |
#api_key=GEMINI_API_KEY,
|
105 |
temperature=0.1,
|
106 |
-
max_new_tokens=
|
107 |
timeout=60,
|
108 |
)
|
109 |
|
|
|
99 |
"""Build the graph"""
|
100 |
# First create the HuggingFaceEndpoint
|
101 |
llm_endpoint = HuggingFaceEndpoint(
|
102 |
+
repo_id="mistralai/Mistral-7B-Instruct-v0.2",
|
103 |
huggingfacehub_api_token=HUGGINGFACEHUB_API_TOKEN,
|
104 |
#api_key=GEMINI_API_KEY,
|
105 |
temperature=0.1,
|
106 |
+
max_new_tokens=1024,
|
107 |
timeout=60,
|
108 |
)
|
109 |
|