alisamak commited on
Commit
44e52fd
·
verified ·
1 Parent(s): bd3446c
Files changed (1) hide show
  1. 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
- model_kwargs={"max_new_tokens": 512}
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)