From 724cec30e96557fc6e30bbce08fa2ef4c10c32fd Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Thu, 16 Jan 2025 00:56:56 -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 9a97015f9b..e52ac2cbf0 100644 --- 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)