From 1dfb9e9505889085c692c63b8258d982a387bd0d Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Tue, 30 Dec 2025 09:19:32 +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 | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/unsloth/models/sentence_transformer.py b/unsloth/models/sentence_transformer.py index 54cb5ad846..9be5b8d229 100644 --- a/unsloth/models/sentence_transformer.py +++ b/unsloth/models/sentence_transformer.py @@ -106,6 +106,7 @@ class FastSentenceTransformer(FastModel): Supports transformers 4. """ from transformers.models.mpnet import modeling_mpnet + # add supports_gradient_checkpointing flag modeling_mpnet.MPNetModel.supports_gradient_checkpointing = True @@ -200,6 +201,7 @@ class FastSentenceTransformer(FastModel): Supports transformers 5. """ from transformers.models.mpnet import modeling_mpnet + # add supports_gradient_checkpointing flag modeling_mpnet.MPNetModel.supports_gradient_checkpointing = True @@ -700,8 +702,10 @@ class FastSentenceTransformer(FastModel): ) if not has_modules_json and load_in_4bit: - print("Unsloth: No modules.json found. This is not a sentence-transformers model.\n" - "Forcing 16-bit loading to simplify merged model saving.") + print( + "Unsloth: No modules.json found. This is not a sentence-transformers model.\n" + "Forcing 16-bit loading to simplify merged model saving." + ) load_in_4bit = False load_in_16bit = True