From 0e8b7339af0db5ee21ddd34cec8377fb09830c89 Mon Sep 17 00:00:00 2001
From: Awni Hannun <awni@apple.com>
Date: Fri, 8 Nov 2024 15:17:37 -0800
Subject: [PATCH] fix

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

diff --git a/llms/mlx_lm/models/hunyuan.py b/llms/mlx_lm/models/hunyuan.py
index d36b600f..b098c20d 100644
--- a/llms/mlx_lm/models/hunyuan.py
+++ b/llms/mlx_lm/models/hunyuan.py
@@ -269,7 +269,7 @@ class Model(nn.Module):
         cache=None,
     ):
         out = self.model(inputs, cache)
-        return self.embed_tokens.as_linear(out)
+        return self.model.embed_tokens.as_linear(out)
 
     def sanitize(self, weights):
         if "model.layers.0.mlp.experts.0.up_proj.weight" not in weights: