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: