bug fig on last step processing

model.py

@@ -177,6 +177,10 @@ class DAC(PreTrainedModel):
             chunk = zq[:,:, start:end]  # (b, d, chunk_size)
             waveform = self.dac.decode(chunk.to(self.device))  # (b, 1, chunk_size*self.downsampling_rate)
             waveform = waveform.cpu()
+
+            waveform_len = waveform.shape[-1]
+            if waveform_len < overlap_size_in_data_space:
+                overlap_size_in_data_space = waveform_len
             
             if isinstance(waveform_concat, type(None)):
                 waveform_concat = waveform.clone()