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