fix: decode chunksize for audiovae_v2

2026-04-08 16:31:36 +08:00
parent 8f95d13073
commit 82d77d445c
2 changed files with 10 additions and 8 deletions
@@ -436,6 +436,7 @@ class AudioVAE(nn.Module):
        self.out_sample_rate = out_sample_rate
        self.sr_bin_boundaries = sr_bin_boundaries
        self.chunk_size = math.prod(encoder_rates)
+        self.decode_chunk_size = math.prod(decoder_rates)

    def preprocess(self, audio_data, sample_rate):
        if sample_rate is None: