Merge pull request #114 from ace-step/short_duration_for_decode_overlap

Short duration for decode overlap
2025-05-10 20:52:27 +08:00
parent 265ee5818d 1da4ba715e
commit 7da27aa311
1 changed files with 1 additions and 1 deletions
@@ -1490,7 +1490,7 @@ class ACEStepPipeline:
        audio_lengths = [target_wav_duration_second * sample_rate] * bs
        pred_latents = latents
        with torch.no_grad():
-            if self.overlapped_decode:
+            if self.overlapped_decode and target_wav_duration_second > 48:
                _, pred_wavs = self.music_dcae.decode_overlap(pred_latents, sr=sample_rate)
            else:
                _, pred_wavs = self.music_dcae.decode(pred_latents, sr=sample_rate)