gordonhubackup commited on
Commit
3c1bf85
1 Parent(s): 28e0ff2

faster setup

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -49,7 +49,7 @@ def parse_args():
49
  parser.add_argument("--num_beams", type=int, default=1)
50
  parser.add_argument("--max_new_tokens", type=int, default=512)
51
  parser.add_argument("--num-visual-tokens", type=int, default=256)
52
- parser.add_argument("--gpu-id", type=int, default=4)
53
  args = parser.parse_args()
54
  return args
55
 
 
49
  parser.add_argument("--num_beams", type=int, default=1)
50
  parser.add_argument("--max_new_tokens", type=int, default=512)
51
  parser.add_argument("--num-visual-tokens", type=int, default=256)
52
+ parser.add_argument("--gpu-id", type=int, default=0)
53
  args = parser.parse_args()
54
  return args
55