Spaces:
Runtime error
Runtime error
fix
Browse files- LG_agent.py +1 -1
LG_agent.py
CHANGED
|
@@ -9,7 +9,7 @@ from langgraph.prebuilt import tools_condition
|
|
| 9 |
llm_endpoint = HuggingFaceEndpoint(
|
| 10 |
repo_id="google/flan-t5-small",
|
| 11 |
temperature=0,
|
| 12 |
-
|
| 13 |
)
|
| 14 |
|
| 15 |
chat = ChatHuggingFace(llm=llm_endpoint, verbose=True)
|
|
|
|
| 9 |
llm_endpoint = HuggingFaceEndpoint(
|
| 10 |
repo_id="google/flan-t5-small",
|
| 11 |
temperature=0,
|
| 12 |
+
max_new_tokens=512
|
| 13 |
)
|
| 14 |
|
| 15 |
chat = ChatHuggingFace(llm=llm_endpoint, verbose=True)
|