Method __init__

src/voxcpm/modules/audiovae/audio_vae_v2.py:163–165

(self, dim)

class NoiseBlock(nn.Module):
    def __init__(self, dim):
        super().__init__()
        self.linear = WNCausalConv1d(dim, dim, kernel_size=1, bias=False)

    def forward(self, x):
        B, C, T = x.shape