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