fffiloni commited on
Commit
3781900
·
verified ·
1 Parent(s): 31eb880

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -124,8 +124,8 @@ def infer(prompt, cond_image_path, cond_audio_path):
124
  "--ckpt_dir", "weights/Wan2.1-I2V-14B-480P",
125
  "--wav2vec_dir", "weights/chinese-wav2vec2-base",
126
  "--input_json", "./examples/single_example_1.json",
127
- "--sample_steps", "4",
128
- "--num_persistent_param_in_dit", str(0),
129
  "--mode", "streaming",
130
  "--use_teacache",
131
  "--save_file", "multi_long_mediumvram_exp"
 
124
  "--ckpt_dir", "weights/Wan2.1-I2V-14B-480P",
125
  "--wav2vec_dir", "weights/chinese-wav2vec2-base",
126
  "--input_json", "./examples/single_example_1.json",
127
+ "--sample_steps", "2",
128
+ "--num_persistent_param_in_dit", str(USED_VRAM_PARAMS),
129
  "--mode", "streaming",
130
  "--use_teacache",
131
  "--save_file", "multi_long_mediumvram_exp"