def __init__(self, dims, multiplxer=4):
    super(MLP, self).__init__()
    hidden = int((dims * multiplxer))
    self.out = nn.Sequential(nn.Linear(dims, hidden), nn.GELU(), nn.Linear(hidden, dims))
def __init__(self, dims, multiplxer=4):
    super(MLP, self).__init__()
    hidden = int((dims * multiplxer))
    self.out = nn.Sequential(nn.Linear(dims, hidden), nn.GELU(), nn.Linear(hidden, dims))