Method __init__

diffsynth/models/stepvideo_vae.py:545–554

(self,
        in_channels,
        with_conv,
        stride
    )

class Downsample3D(nn.Module):
    def __init__(self,
        in_channels,
        with_conv,
        stride
    ):
        super().__init__()

        self.with_conv = with_conv
        if with_conv:
            self.conv = CausalConv(in_channels, in_channels, kernel_size=3, stride=stride)

    def forward(self, x, is_init=True):
        if self.with_conv: