nyarunyarunya commited on
Commit
345422a
·
verified ·
1 Parent(s): 23a69a1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -100,13 +100,13 @@ def _parse_args():
100
  parser.add_argument(
101
  "--task",
102
  type=str,
103
- default="t2v-14B",
104
  choices=list(WAN_CONFIGS.keys()),
105
  help="The task to run.")
106
  parser.add_argument(
107
  "--size",
108
  type=str,
109
- default="1280*720",
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=None,
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",