Commit 381e62dd authored by smcio's avatar smcio
Browse files

Modify default SoundStream strides to 480 to match default 24kHz sampling rate

parent 9733c558

audiolm_pytorch/soundstream.py

+1 −1
Original line number Diff line number Diff line
@@ -343,7 +343,7 @@ class SoundStream(nn.Module): 
        self,

        *,

        channels = 32,

        strides = (2, 4, 5, 8),

        strides = (3, 4, 5, 8),

        channel_mults = (2, 4, 8, 16),

        codebook_dim = 512,

        codebook_size = 1024,