debisoft commited on
Commit
08fff16
·
1 Parent(s): 5280b47

Using deepseek and max_steps=10

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -119,9 +119,9 @@ max_tokens=2096,
119
  temperature=0.5,
120
  #model_id='deepseek-ai/deepseek-coder-7b-instruct-v1.5',# it is possible that this model may be overloaded
121
  #model_id='Qwen/Qwen2.5-Coder-7B-Instruct',
122
- model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
123
  #model_id='Qwen/Qwen2.5-72B-Instruct',
124
- #model_id='deepseek-ai/DeepSeek-R1-Distill-Qwen-32B',
125
  custom_role_conversions=None,
126
  )
127
 
@@ -135,7 +135,7 @@ with open("prompts.yaml", 'r') as stream:
135
  agent = CodeAgent(
136
  model=model,
137
  tools=[fetch_wikipedia_page_content, final_answer, say_hello, image_generation_tool, search_tool, VisitWebpageTool()], ## add your tools here (don't remove final answer)
138
- max_steps=15,
139
  verbosity_level=1,
140
  grammar=None,
141
  planning_interval=None,
 
119
  temperature=0.5,
120
  #model_id='deepseek-ai/deepseek-coder-7b-instruct-v1.5',# it is possible that this model may be overloaded
121
  #model_id='Qwen/Qwen2.5-Coder-7B-Instruct',
122
+ #model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
123
  #model_id='Qwen/Qwen2.5-72B-Instruct',
124
+ model_id='deepseek-ai/DeepSeek-R1-Distill-Qwen-32B',
125
  custom_role_conversions=None,
126
  )
127
 
 
135
  agent = CodeAgent(
136
  model=model,
137
  tools=[fetch_wikipedia_page_content, final_answer, say_hello, image_generation_tool, search_tool, VisitWebpageTool()], ## add your tools here (don't remove final answer)
138
+ max_steps=10,
139
  verbosity_level=1,
140
  grammar=None,
141
  planning_interval=None,