Ali2206 commited on
Commit
fddf521
·
verified ·
1 Parent(s): 1da2cfd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -154,7 +154,7 @@ def init_agent():
154
  step_rag_num=8,
155
  seed=100,
156
  additional_default_tools=[],
157
- device_map="auto"
158
  )
159
  agent.init_model()
160
  return agent
 
154
  step_rag_num=8,
155
  seed=100,
156
  additional_default_tools=[],
157
+
158
  )
159
  agent.init_model()
160
  return agent