def __init__(self, seq, dims):
    super(MixerLayer, self).__init__()
    self.layer_norm1 = nn.LayerNorm(dims)
    self.mlp1 = MLP(seq, multiplxer=0.5)
    self.layer_norm2 = nn.LayerNorm(dims)
    self.mlp2 = MLP(dims)
def __init__(self, seq, dims):
    super(MixerLayer, self).__init__()
    self.layer_norm1 = nn.LayerNorm(dims)
    self.mlp1 = MLP(seq, multiplxer=0.5)
    self.layer_norm2 = nn.LayerNorm(dims)
    self.mlp2 = MLP(dims)