Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -100,13 +100,13 @@ def _parse_args():
|
|
| 100 |
parser.add_argument(
|
| 101 |
"--task",
|
| 102 |
type=str,
|
| 103 |
-
default="
|
| 104 |
choices=list(WAN_CONFIGS.keys()),
|
| 105 |
help="The task to run.")
|
| 106 |
parser.add_argument(
|
| 107 |
"--size",
|
| 108 |
type=str,
|
| 109 |
-
default="
|
| 110 |
choices=list(SIZE_CONFIGS.keys()),
|
| 111 |
help="The area (width*height) of the generated video. For the I2V task, the aspect ratio of the output video will follow that of the input image."
|
| 112 |
)
|
|
@@ -119,7 +119,7 @@ def _parse_args():
|
|
| 119 |
parser.add_argument(
|
| 120 |
"--ckpt_dir",
|
| 121 |
type=str,
|
| 122 |
-
default=
|
| 123 |
help="The path to the checkpoint directory.")
|
| 124 |
parser.add_argument(
|
| 125 |
"--offload_model",
|
|
|
|
| 100 |
parser.add_argument(
|
| 101 |
"--task",
|
| 102 |
type=str,
|
| 103 |
+
default="i2v-14B",
|
| 104 |
choices=list(WAN_CONFIGS.keys()),
|
| 105 |
help="The task to run.")
|
| 106 |
parser.add_argument(
|
| 107 |
"--size",
|
| 108 |
type=str,
|
| 109 |
+
default="960*544",
|
| 110 |
choices=list(SIZE_CONFIGS.keys()),
|
| 111 |
help="The area (width*height) of the generated video. For the I2V task, the aspect ratio of the output video will follow that of the input image."
|
| 112 |
)
|
|
|
|
| 119 |
parser.add_argument(
|
| 120 |
"--ckpt_dir",
|
| 121 |
type=str,
|
| 122 |
+
default="/root/weight",
|
| 123 |
help="The path to the checkpoint directory.")
|
| 124 |
parser.add_argument(
|
| 125 |
"--offload_model",
|