Spaces:
Sleeping
Sleeping
gordonhubackup
commited on
Commit
•
3c1bf85
1
Parent(s):
28e0ff2
faster setup
Browse files
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=
|
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 |
|