From d4e688edc3e4a25e31c3ec6fe340019e8ffdc52f Mon Sep 17 00:00:00 2001
From: Shunta Saito <shunta.saito@gmail.com>
Date: Sat, 22 Feb 2025 19:55:42 +0900
Subject: [PATCH] Fix reference to layers

---
 llms/mlx_lm/models/plamo2.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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