fffiloni commited on
Commit
ca4adb8
·
verified ·
1 Parent(s): 71be38c

Update inference_from_video.py

Browse files
Files changed (1) hide show
  1. 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