Nyandwi commited on
Commit
cb9c64c
·
verified ·
1 Parent(s): 3639a12

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -712,7 +712,7 @@ if __name__ == "__main__":
712
  )
713
  # argparser.add_argument("--model-path", type=str, default="facebook/opt-350m")
714
  argparser.add_argument("--model-base", type=str, default=None)
715
- argparser.add_argument("--device_map", type=str, default="cpu")
716
  # argparser.add_argument("--num-gpus", type=int, default=1)
717
  argparser.add_argument("--conv-mode", type=str, default=None)
718
  argparser.add_argument("--temperature", type=float, default=0.7)
 
712
  )
713
  # argparser.add_argument("--model-path", type=str, default="facebook/opt-350m")
714
  argparser.add_argument("--model-base", type=str, default=None)
715
+ argparser.add_argument("--device_map", type=str, default="auto")
716
  # argparser.add_argument("--num-gpus", type=int, default=1)
717
  argparser.add_argument("--conv-mode", type=str, default=None)
718
  argparser.add_argument("--temperature", type=float, default=0.7)