diff --git a/unsloth/models/mistral.py b/unsloth/models/mistral.py index 4d91e98371..0c13f998f7 100644 --- a/unsloth/models/mistral.py +++ b/unsloth/models/mistral.py @@ -14,7 +14,7 @@ from .llama import * -from transformers.models.llama.modeling_mistral import ( +from transformers.models.mistral.modeling_mistral import ( MistralAttention, MistralDecoderLayer, MistralModel,