diff --git a/llms/mlx_lm/models/plamo2.py b/llms/mlx_lm/models/plamo2.py index c6e413b0..f9a0c40c 100644 --- a/llms/mlx_lm/models/plamo2.py +++ b/llms/mlx_lm/models/plamo2.py @@ -1798,4 +1798,4 @@ class Model(PlamoPreTrainedModel): @property def layers(self): - return self.model.layers + return self.model.layers.layers