wt002 commited on
Commit
49248c8
·
verified ·
1 Parent(s): d0ef80b

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -1
agent.py CHANGED
@@ -467,7 +467,7 @@ from langchain_huggingface import HuggingFaceEndpoint
467
  # Initialize the HuggingFaceEndpoint with the desired model and parameters
468
  llm = HuggingFaceEndpoint(
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
 
467
  # Initialize the HuggingFaceEndpoint with the desired model and parameters
468
  llm = HuggingFaceEndpoint(
469
  repo_id="Qwen/Qwen2.5-Coder-32B-Instruct",
470
+ task="conversational",
471
  huggingfacehub_api_token=hf_token,
472
  temperature=0.7,
473
  max_new_tokens=1024