Method __init__

models/transformer/wan/modules/t2m_model.py:251–252

(self, dim, eps=1e-6, elementwise_affine=False)

class WanLayerNorm(nn.LayerNorm):

    def __init__(self, dim, eps=1e-6, elementwise_affine=False):
        super().__init__(dim, elementwise_affine=elementwise_affine, eps=eps)

    def forward(self, x):
        return super().forward(x.float()).type_as(x)