Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -405,7 +405,7 @@ if __name__ == "__main__":
|
|
| 405 |
parser.add_argument("--model-list-mode", type=str, default="once",
|
| 406 |
choices=["once", "reload"])
|
| 407 |
parser.add_argument("--model-path", type=str, default="mPLUG/DocOwl1.5-Omni")
|
| 408 |
-
parser.add_argument("--device", type=str, default="cuda")
|
| 409 |
parser.add_argument("--load-8bit", action="store_true")
|
| 410 |
parser.add_argument("--load-4bit", action="store_true")
|
| 411 |
parser.add_argument("--moderate", action="store_true")
|
|
|
|
| 405 |
parser.add_argument("--model-list-mode", type=str, default="once",
|
| 406 |
choices=["once", "reload"])
|
| 407 |
parser.add_argument("--model-path", type=str, default="mPLUG/DocOwl1.5-Omni")
|
| 408 |
+
parser.add_argument("--device", type=str, default="cuda:0")
|
| 409 |
parser.add_argument("--load-8bit", action="store_true")
|
| 410 |
parser.add_argument("--load-4bit", action="store_true")
|
| 411 |
parser.add_argument("--moderate", action="store_true")
|