diff --git a/unsloth/models/sentence_transformer.py b/unsloth/models/sentence_transformer.py index 8f438e1a9b..3c4a221b00 100644 --- a/unsloth/models/sentence_transformer.py +++ b/unsloth/models/sentence_transformer.py @@ -252,7 +252,7 @@ class FastSentenceTransformer(FastModel): hidden_states, attention_mask, position_bias, - use_reentrant = True, # required for torch >= 2.9 + use_reentrant = True, # required for torch >= 2.9 ) else: # original code from here on