From cfcf2a75ffadd8c2a80f2c2e157ce2bfabbc0d9a 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 11:26:28 +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 | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/unsloth/models/sentence_transformer.py b/unsloth/models/sentence_transformer.py index b2859c8644..37b8afb35b 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 @@ -762,7 +766,8 @@ class FastSentenceTransformer(FastModel): # check which adapter files exist before save_pretrained adapter_files = ["adapter_model.safetensors", "adapter_config.json"] existing_before = { - f for f in adapter_files + f + for f in adapter_files if os.path.exists(os.path.join(save_directory, f)) }