Spaces:
Running
on
Zero
Running
on
Zero
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",
|