From f8c451e22a3c97861ecb39f026f7067d7aba6e5f Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Wed, 17 Dec 2025 11:39:07 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- unsloth/models/sentence_transformer.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/unsloth/models/sentence_transformer.py b/unsloth/models/sentence_transformer.py index e94d4f0c60..89ac4088bc 100644 --- a/unsloth/models/sentence_transformer.py +++ b/unsloth/models/sentence_transformer.py @@ -120,7 +120,7 @@ class FastSentenceTransformer(FastModel): ): # backwards compatibility for older transformers versions (4.57.3 and below) head_mask = kwargs.pop("head_mask", None) - + position_bias = self.compute_position_bias(hidden_states) all_hidden_states = () if output_hidden_states else None all_attentions = () if output_attentions else None @@ -138,6 +138,7 @@ class FastSentenceTransformer(FastModel): return module( *inputs, output_attentions = output_attentions ) + return custom_forward layer_outputs = torch.utils.checkpoint.checkpoint(