Shago commited on
Commit
9cb3af7
·
verified ·
1 Parent(s): f6de0d4

Update agents/agents_nodes.py

Browse files
Files changed (1) hide show
  1. agents/agents_nodes.py +6 -6
agents/agents_nodes.py CHANGED
@@ -26,12 +26,12 @@ 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="huggingface",
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
+ provider="huggingface",
35
  )
36
 
37
  llm = ChatHuggingFace(llm=llm_endpoint)