anirbans403 commited on
Commit
4ad0811
·
1 Parent(s): 30104af

HuggingFace RepoId

Browse files
Files changed (1) hide show
  1. agent.py +1 -3
agent.py CHANGED
@@ -162,11 +162,9 @@ def build_graph(provider: str = "huggingface"):
162
  # Groq https://console.groq.com/docs/models
163
  llm = ChatGroq(model="qwen-qwq-32b", temperature=0) # optional : qwen-qwq-32b gemma2-9b-it
164
  elif provider == "huggingface":
165
- # TODO: Add huggingface endpoint
166
  llm = ChatHuggingFace(
167
  llm=HuggingFaceEndpoint(
168
- url="https://api-inference.huggingface.co/models/Meta-DeepLearning/llama-2-7b-chat-hf",
169
- temperature=0,
170
  ),
171
  )
172
  else:
 
162
  # Groq https://console.groq.com/docs/models
163
  llm = ChatGroq(model="qwen-qwq-32b", temperature=0) # optional : qwen-qwq-32b gemma2-9b-it
164
  elif provider == "huggingface":
 
165
  llm = ChatHuggingFace(
166
  llm=HuggingFaceEndpoint(
167
+ repo_id = "meta-llama/Llama-2-7b-chat-hf"
 
168
  ),
169
  )
170
  else: