mtmx commited on
Commit
867aba7
·
verified ·
1 Parent(s): 5f6cc84

change model to phi-3 mini, qwen is overloading

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -171,7 +171,7 @@ final_answer = FinalAnswerTool()
171
  model = HfApiModel(
172
  max_tokens=2096,
173
  temperature=0.5,
174
- model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
175
  custom_role_conversions=None,
176
  )
177
 
@@ -184,7 +184,10 @@ with open("prompts.yaml", 'r') as stream:
184
 
185
  agent = CodeAgent(
186
  model=model,
187
- tools=[final_answer, image_generation_tool], ## add your tools here (don't remove final answer)
 
 
 
188
  max_steps=6,
189
  verbosity_level=1,
190
  grammar=None,
 
171
  model = HfApiModel(
172
  max_tokens=2096,
173
  temperature=0.5,
174
+ model_id='microsoft/Phi-3-mini-4k-instruct',
175
  custom_role_conversions=None,
176
  )
177
 
 
184
 
185
  agent = CodeAgent(
186
  model=model,
187
+ tools=[final_answer,
188
+ image_generation_tool,
189
+ roast_ai_frameworks_meme,
190
+ get_current_time_in_timezone]
191
  max_steps=6,
192
  verbosity_level=1,
193
  grammar=None,