Spaces:
Sleeping
Sleeping
Update inference_from_video.py
Browse files- inference_from_video.py +4 -0
inference_from_video.py
CHANGED
@@ -182,6 +182,10 @@ def main():
|
|
182 |
latents = model.inference(scheduler, None, prompt, None, num_steps, guidance, num_samples, disable_progress=True, device=device)
|
183 |
mel = vae.decode_first_stage(latents)
|
184 |
wave = vae.decode_to_waveform(mel)
|
|
|
|
|
|
|
|
|
185 |
|
186 |
all_outputs += [item for item in wave]
|
187 |
|
|
|
182 |
latents = model.inference(scheduler, None, prompt, None, num_steps, guidance, num_samples, disable_progress=True, device=device)
|
183 |
mel = vae.decode_first_stage(latents)
|
184 |
wave = vae.decode_to_waveform(mel)
|
185 |
+
|
186 |
+
# Ensure the waveform is exactly 8 seconds long
|
187 |
+
num_samples_8_seconds = sample_rate * max_len_in_seconds
|
188 |
+
wave = [wav[:num_samples_8_seconds] for wav in wave]
|
189 |
|
190 |
all_outputs += [item for item in wave]
|
191 |
|