wt002 commited on
Commit
d0ef80b
·
verified ·
1 Parent(s): 25cf5ed

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +3 -1
agent.py CHANGED
@@ -469,10 +469,12 @@ llm = HuggingFaceEndpoint(
469
  repo_id="Qwen/Qwen2.5-Coder-32B-Instruct",
470
  task="text-generation",
471
  huggingfacehub_api_token=hf_token,
472
- model_kwargs={"temperature": 0.7, "max_new_tokens": 1024}
 
473
  )
474
 
475
 
 
476
  # Initialize the LangChain agent with the tool(s) and the model
477
  agent = initialize_agent(
478
  tools=tools,
 
469
  repo_id="Qwen/Qwen2.5-Coder-32B-Instruct",
470
  task="text-generation",
471
  huggingfacehub_api_token=hf_token,
472
+ temperature=0.7,
473
+ max_new_tokens=1024
474
  )
475
 
476
 
477
+
478
  # Initialize the LangChain agent with the tool(s) and the model
479
  agent = initialize_agent(
480
  tools=tools,