EtienneB commited on
Commit
d55b1b4
·
1 Parent(s): 263c392

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +3 -1
agent.py CHANGED
@@ -143,7 +143,9 @@ def build_graph():
143
  """Build the graph"""
144
  # First create the HuggingFaceEndpoint
145
  llm_endpoint = HuggingFaceEndpoint(
146
- repo_id="Qwen/Qwen2.5-14B-Instruct",
 
 
147
  huggingfacehub_api_token=HUGGINGFACEHUB_API_TOKEN,
148
  temperature=0.1, # Maximum determinism
149
  max_new_tokens=128, # Even more restrictive
 
143
  """Build the graph"""
144
  # First create the HuggingFaceEndpoint
145
  llm_endpoint = HuggingFaceEndpoint(
146
+ # repo_id="Qwen/Qwen2.5-14B-Instruct",
147
+ # repo_id="HuggingFaceH4/mistral-7b-anthropic",
148
+ repo_id="Qwen/Qwen2.5-Coder-32B-Instruct",
149
  huggingfacehub_api_token=HUGGINGFACEHUB_API_TOKEN,
150
  temperature=0.1, # Maximum determinism
151
  max_new_tokens=128, # Even more restrictive