Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -712,7 +712,7 @@ if __name__ == "__main__":
|
|
712 |
)
|
713 |
# argparser.add_argument("--model-path", type=str, default="facebook/opt-350m")
|
714 |
argparser.add_argument("--model-base", type=str, default=None)
|
715 |
-
argparser.add_argument("--device_map", type=str, default="
|
716 |
# argparser.add_argument("--num-gpus", type=int, default=1)
|
717 |
argparser.add_argument("--conv-mode", type=str, default=None)
|
718 |
argparser.add_argument("--temperature", type=float, default=0.7)
|
|
|
712 |
)
|
713 |
# argparser.add_argument("--model-path", type=str, default="facebook/opt-350m")
|
714 |
argparser.add_argument("--model-base", type=str, default=None)
|
715 |
+
argparser.add_argument("--device_map", type=str, default="auto")
|
716 |
# argparser.add_argument("--num-gpus", type=int, default=1)
|
717 |
argparser.add_argument("--conv-mode", type=str, default=None)
|
718 |
argparser.add_argument("--temperature", type=float, default=0.7)
|