Spaces:
Runtime error
Runtime error
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 |
|