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