Commit
·
00255ab
1
Parent(s):
36f7b8e
sdf
Browse files
agent.py
CHANGED
@@ -169,15 +169,15 @@ def build_graph(provider: str = "huggingface"):
|
|
169 |
repo_id = "togethercomputer/evo-1-131k-base"
|
170 |
if not hf_token:
|
171 |
raise ValueError("HF_TOKEN environment variable not set. It's required for Hugging Face provider.")
|
172 |
-
llm =
|
173 |
repo_id="HuggingFaceH4/zephyr-7b-beta",
|
174 |
task="text-generation",
|
175 |
max_new_tokens=1000,
|
176 |
do_sample=False,
|
177 |
repetition_penalty=1.03,
|
178 |
-
|
179 |
)
|
180 |
-
llm = ChatHuggingFace(llm=llm,
|
181 |
else:
|
182 |
raise ValueError("Invalid provider. Choose 'google', 'groq' or 'huggingface'.")
|
183 |
# Bind tools to LLM
|
|
|
169 |
repo_id = "togethercomputer/evo-1-131k-base"
|
170 |
if not hf_token:
|
171 |
raise ValueError("HF_TOKEN environment variable not set. It's required for Hugging Face provider.")
|
172 |
+
llm = HuggingFaceEndpoint(
|
173 |
repo_id="HuggingFaceH4/zephyr-7b-beta",
|
174 |
task="text-generation",
|
175 |
max_new_tokens=1000,
|
176 |
do_sample=False,
|
177 |
repetition_penalty=1.03,
|
178 |
+
|
179 |
)
|
180 |
+
llm = ChatHuggingFace(llm=llm,recursion_limit=50)
|
181 |
else:
|
182 |
raise ValueError("Invalid provider. Choose 'google', 'groq' or 'huggingface'.")
|
183 |
# Bind tools to LLM
|