Update mistral.py

2025-01-16 00:58:46 -08:00 · 2025-01-16 00:58:46 -08:00 · 8ed010b4a5
commit 8ed010b4a5
parent 724cec30e9
1 changed files with 1 additions and 0 deletions
--- 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: