wt002 commited on
Commit
2ca29a4
·
verified ·
1 Parent(s): 02f4c11

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -1
agent.py CHANGED
@@ -472,7 +472,7 @@ llm = HuggingFaceEndpoint(
472
  task="text-generation",
473
  huggingfacehub_api_token=os.getenv("HF_TOKEN"),
474
  temperature=0.7,
475
- max_length=1024
476
  )
477
 
478
  # Initialize the LangChain agent with the tool(s) and the model
 
472
  task="text-generation",
473
  huggingfacehub_api_token=os.getenv("HF_TOKEN"),
474
  temperature=0.7,
475
+ max_new_tokens=1024
476
  )
477
 
478
  # Initialize the LangChain agent with the tool(s) and the model