diff --git a/llms/mlx_lm/models/cohere2.py b/llms/mlx_lm/models/cohere2.py index c6f3e885..a078409b 100644 --- a/llms/mlx_lm/models/cohere2.py +++ b/llms/mlx_lm/models/cohere2.py @@ -29,7 +29,6 @@ class ModelArgs(BaseModelArgs): class LayerNorm2D(nn.Module): - def __init__(self, d1, d2, eps): super().__init__() self.weight = mx.zeros((d1, d2))