wishwakankanamg commited on
Commit
00255ab
·
1 Parent(s): 36f7b8e
Files changed (1) hide show
  1. agent.py +3 -3
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 = HuggingFacePipeline.from_model_id(
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
 
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