Update mistral.py

2025-01-16 00:56:56 -08:00 · 2025-01-16 00:56:56 -08:00 · 724cec30e9
commit 724cec30e9
parent e3a2dbbab6
1 changed files with 1 additions and 0 deletions
--- a/unsloth/models/mistral.py
+++ b/unsloth/models/mistral.py
@ -306,6 +306,7 @@ class FastMistralModel(FastLlamaModel):
        # Just for Mistral Nemo models!
        if function is not None:
            function = patch_mistral_nemo_attention(function)
+            print(function)
            # if True:#init_name is not None:
            exec(function, globals())
            MistralAttention.__init__  = eval(init_name)