wt002 commited on
Commit
94dcdd5
·
verified ·
1 Parent(s): 9eabe16

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +5 -3
agent.py CHANGED
@@ -468,10 +468,12 @@ from langchain_huggingface import HuggingFaceEndpoint
468
 
469
  # Initialize the HuggingFaceEndpoint with the desired model and parameters
470
  llm = HuggingFaceEndpoint(
471
- repo_id="meta-llama/Meta-Llama-3-70B-Instruct",
472
  task="text-generation",
473
- huggingfacehub_api_token=hf_token,
474
- model_kwargs={"temperature": 0.7, "max_length": 1024}
 
 
475
  )
476
 
477
 
 
468
 
469
  # Initialize the HuggingFaceEndpoint with the desired model and parameters
470
  llm = HuggingFaceEndpoint(
471
+ repo_id="Qwen/Qwen2.5-Coder-32B-Instruct",
472
  task="text-generation",
473
+ huggingfacehub_api_token=os.getenv("HF_TOKEN"),
474
+ temperature=0.7,
475
+ model_kwargs={"max_length": 1024}
476
+ )
477
  )
478
 
479