add vram optimization cmdline args

2025-05-09 22:47:41 +08:00
parent 56ae032172
commit f23b7b34e3
3 changed files with 13 additions and 6 deletions
@@ -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 = []