From 8ed010b4a5d6b9588fd837149fc06fdc3dbb88f2 Mon Sep 17 00:00:00 2001
From: Daniel Han <danielhanchen@gmail.com>
Date: Thu, 16 Jan 2025 00:58:46 -0800
Subject: [PATCH] Update mistral.py

---
 unsloth/models/mistral.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/unsloth/models/mistral.py b/unsloth/models/mistral.py
index e52ac2cbf0..4edc3b7990 100644
--- a/unsloth/models/mistral.py
+++ b/unsloth/models/mistral.py
@@ -305,6 +305,7 @@ class FastMistralModel(FastLlamaModel):
         )
         # Just for Mistral Nemo models!
         if function is not None:
+            print(function)
             function = patch_mistral_nemo_attention(function)
             print(function)
             # if True:#init_name is not None: