Spaces:
Running
on
L40S
Running
on
L40S
Update app.py
Browse files
app.py
CHANGED
@@ -126,7 +126,7 @@ def infer(prompt, cond_image_path, cond_audio_path):
|
|
126 |
"--wav2vec_dir", "weights/chinese-wav2vec2-base",
|
127 |
"--input_json", "./examples/single_example_1.json",
|
128 |
"--sample_steps", "20",
|
129 |
-
"--num_persistent_param_in_dit", USED_VRAM_PARAMS,
|
130 |
"--mode", "streaming",
|
131 |
"--use_teacache",
|
132 |
"--save_file", "multi_long_mediumvram_exp"
|
|
|
126 |
"--wav2vec_dir", "weights/chinese-wav2vec2-base",
|
127 |
"--input_json", "./examples/single_example_1.json",
|
128 |
"--sample_steps", "20",
|
129 |
+
"--num_persistent_param_in_dit", str(USED_VRAM_PARAMS),
|
130 |
"--mode", "streaming",
|
131 |
"--use_teacache",
|
132 |
"--save_file", "multi_long_mediumvram_exp"
|