bug fix

model.py

@@ -193,7 +193,10 @@ class DAC(PreTrainedModel):
                     rest_of_new_x = waveform[:,:,overlap_size_in_data_space:]
                     overlap_x_from_prev_x = waveform_concat[:,:,-overlap_size_in_data_space:]  # (b, 1, overlap_size_in_data_space)
                     overlap_x_from_new_x = waveform[:,:,:overlap_size_in_data_space]  # (b, 1, overlap_size_in_data_space)
-                    overlap = (overlap_x_from_prev_x + overlap_x_from_new_x) / 2  # take mean; maybe there's a better strategy but it seems to work fine.
+                    if not overlap_x_from_new_x.shape[-1] == 0:
+                        overlap = (overlap_x_from_prev_x + overlap_x_from_new_x) / 2  # take mean; maybe there's a better strategy but it seems to work fine.
+                    else:
+                        overlap = overlap_x_from_prev_x
                     waveform_concat = torch.cat((prev_x, overlap, rest_of_new_x), dim=-1)  # (b, 1, ..)
                 else:
                     prev_x = waveform_concat