Shago commited on
Commit
4d37ed9
·
verified ·
1 Parent(s): 87c927e

Update agents/agents_nodes.py

Browse files
Files changed (1) hide show
  1. agents/agents_nodes.py +5 -6
agents/agents_nodes.py CHANGED
@@ -26,12 +26,11 @@ from langchain_huggingface import ChatHuggingFace, HuggingFaceEndpoint
26
 
27
  repo_id= "google/gemma-3n-e4b-it" #"deepseek-ai/DeepSeek-R1-0528-Qwen3-8B" #"google/gemma-2b-it"
28
 
29
- llm_endpoint = HuggingFaceEndpoint(
30
- repo_id=repo_id,
31
- huggingfacehub_api_token=os.environ["HF_TOKEN"],
32
- max_new_tokens=1024,
33
- temperature=0.0,
34
- provider="auto",
35
  )
36
 
37
  llm = ChatHuggingFace(llm=llm_endpoint)
 
26
 
27
  repo_id= "google/gemma-3n-e4b-it" #"deepseek-ai/DeepSeek-R1-0528-Qwen3-8B" #"google/gemma-2b-it"
28
 
29
+ llm_endpoint = HuggingFaceEndpoint(
30
+ repo_id=repo_id,
31
+ huggingfacehub_api_token=os.environ["HF_TOKEN"],
32
+ max_new_tokens=1024,
33
+ temperature=0.0
 
34
  )
35
 
36
  llm = ChatHuggingFace(llm=llm_endpoint)