diff --git a/unsloth/models/sentence_transformer.py b/unsloth/models/sentence_transformer.py index 09236e5a37..98f6dc3a3a 100644 --- a/unsloth/models/sentence_transformer.py +++ b/unsloth/models/sentence_transformer.py @@ -1157,7 +1157,7 @@ class FastSentenceTransformer(FastModel): # Prepare for k-bit training if quantized if is_quantized: - from peft import prepare_model_for_kbit_training + from unsloth.models._utils import prepare_model_for_kbit_training _gc_for_kbit = ( use_gradient_checkpointing