TobDeBer commited on
Commit
cd569ce
·
verified ·
1 Parent(s): 43d6832

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -355,7 +355,7 @@ if __name__ == "__main__":
355
  parser = argparse.ArgumentParser()
356
  parser.add_argument("--share", action="store_true", default=False, help="share gradio app")
357
  parser.add_argument("--port", type=int, default=7860, help="gradio server port")
358
- parser.add_argument("--device", type=str, default="cuda", help="device to run model")
359
  parser.add_argument("--batch", type=int, default=8, help="batch size")
360
  parser.add_argument("--max-gen", type=int, default=1024, help="max")
361
  opt = parser.parse_args()
 
355
  parser = argparse.ArgumentParser()
356
  parser.add_argument("--share", action="store_true", default=False, help="share gradio app")
357
  parser.add_argument("--port", type=int, default=7860, help="gradio server port")
358
+ parser.add_argument("--device", type=str, default="cpu", help="device to run model")
359
  parser.add_argument("--batch", type=int, default=8, help="batch size")
360
  parser.add_argument("--max-gen", type=int, default=1024, help="max")
361
  opt = parser.parse_args()