HCho commited on
Commit
1bf2c7d
·
verified ·
1 Parent(s): 8ee0e1c

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
agent.py CHANGED
@@ -111,8 +111,8 @@ def build_graph(provider: str = "huggingface"):
111
  # TODO: Add huggingface endpoint
112
  llm = ChatHuggingFace(
113
  llm=HuggingFaceEndpoint(
114
- url="https://api-inference.huggingface.co/models/Meta-DeepLearning/llama-2-7b-chat-hf",
115
- temperature=0,
116
  huggingfacehub_api_token=hf_api_key
117
  ),
118
  )
 
111
  # TODO: Add huggingface endpoint
112
  llm = ChatHuggingFace(
113
  llm=HuggingFaceEndpoint(
114
+ endpoint_url="https://api-inference.huggingface.co/models/Meta-DeepLearning/llama-2-7b-chat-hf",
115
+ model_kwargs={"temperature": 0},
116
  huggingfacehub_api_token=hf_api_key
117
  ),
118
  )