updates

60cdc0fd · hungchiayu1 · e8a362f1 · 60cdc0fd
Commit 60cdc0fd authored Dec 29, 2024 by hungchiayu1
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

tangoflux.py tangoflux.py +2 -0

No files found.
--- a/TangoFlux.py
+++ b/TangoFlux.py
@@ -50,6 +50,8 @@ class TangoFluxInference:


            wave = self.vae.decode(latents.transpose(2,1)).sample.cpu()[0]
+        waveform_end = int(duration * self.vae.config.sampling_rate)
+        wave =  wave[:, :, :waveform_end]
        return wave