diff --git a/acestep/pipeline_ace_step.py b/acestep/pipeline_ace_step.py index e10825b..8c19d0f 100644 --- a/acestep/pipeline_ace_step.py +++ b/acestep/pipeline_ace_step.py @@ -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 > 30: _, pred_wavs = self.music_dcae.decode_overlap(pred_latents, sr=sample_rate) else: _, pred_wavs = self.music_dcae.decode(pred_latents, sr=sample_rate)