Shago commited on
Commit
aa2a1df
·
verified ·
1 Parent(s): a8a9435

Update agents/agents_nodes.py

Browse files
Files changed (1) hide show
  1. agents/agents_nodes.py +3 -2
agents/agents_nodes.py CHANGED
@@ -27,10 +27,11 @@ from langchain_huggingface import ChatHuggingFace, HuggingFaceEndpoint
27
  repo_id="google/gemma-2b-it"
28
 
29
  llm_endpoint = HuggingFaceEndpoint(
30
- repo_id=repo_id
31
  huggingfacehub_api_token=os.environ["HF_TOKEN"], # Set in Space secrets
32
  max_new_tokens=500,
33
- temperature=0
 
34
  )
35
 
36
  llm = ChatHuggingFace(llm=llm_endpoint)
 
27
  repo_id="google/gemma-2b-it"
28
 
29
  llm_endpoint = HuggingFaceEndpoint(
30
+ repo_id=repo_id,
31
  huggingfacehub_api_token=os.environ["HF_TOKEN"], # Set in Space secrets
32
  max_new_tokens=500,
33
+ temperature=0,
34
+ provider="auto",
35
  )
36
 
37
  llm = ChatHuggingFace(llm=llm_endpoint)