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")
|