Method __init__

inference/models/mpt.py:60–65

(self, normalized_shape, eps=1e-05, elementwise_affine=True)

class LPLayerNorm(torch.nn.LayerNorm):
    def __init__(self, normalized_shape, eps=1e-05, elementwise_affine=True):
        super().__init__(
            normalized_shape=normalized_shape,
            eps=eps,
            elementwise_affine=elementwise_affine,
        )

    def forward(self, x):
        module_device = x.device