add vram optimization cmdline args

This commit is contained in:
xushengyuan committed 2025-05-09 22:47:41 +08:00
1 parent 56ae032172
commit f23b7b34e3
3 files changed
+13 -6

No files matched your search

@@ -144,6 +144,7 @@ class MusicDCAE(ModelMixin, ConfigMixin, FromOriginalModelMixin):
@torch.no_grad()
def decode_overlap(self, latents, audio_lengths=None, sr=None):
print("Using Overlapped DCAE and Vocoder Decoding.")
latents = latents / self.scale_factor + self.shift_factor
pred_wavs = []