Update agents/agents_nodes.py
Browse files- 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 |
-
|
| 31 |
-
|
| 32 |
-
|
| 33 |
-
|
| 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)
|