EtienneB commited on
Commit
7faf23e
·
1 Parent(s): c4e5a43

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
agent.py CHANGED
@@ -99,11 +99,11 @@ def build_graph():
99
  """Build the graph"""
100
  # First create the HuggingFaceEndpoint
101
  llm_endpoint = HuggingFaceEndpoint(
102
- repo_id="Qwen/Qwen2.5-Coder-32B-Instruct",
103
  huggingfacehub_api_token=HUGGINGFACEHUB_API_TOKEN,
104
  #api_key=GEMINI_API_KEY,
105
  temperature=0.1,
106
- max_new_tokens=4096,
107
  timeout=60,
108
  )
109
 
 
99
  """Build the graph"""
100
  # First create the HuggingFaceEndpoint
101
  llm_endpoint = HuggingFaceEndpoint(
102
+ repo_id="mistralai/Mistral-7B-Instruct-v0.2",
103
  huggingfacehub_api_token=HUGGINGFACEHUB_API_TOKEN,
104
  #api_key=GEMINI_API_KEY,
105
  temperature=0.1,
106
+ max_new_tokens=1024,
107
  timeout=60,
108
  )
109