Method forward

moss_soundeffect_v2/diffsynth/models/wan_video_dit.py:214–215

(self, x, gate, residual)

        super().__init__()

    def forward(self, x, gate, residual):
        return x + gate * residual

class DiTBlock(nn.Module):
    def __init__(self, has_image_input: bool, dim: int, num_heads: int, ffn_dim: int, eps: float = 1e-6):