From b6f65d1aa58f4b43f65ae1da0b54fc28e64d914a Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Sat, 20 Dec 2025 09:04:13 +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 | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/unsloth/models/sentence_transformer.py b/unsloth/models/sentence_transformer.py index 191be2adbe..abe278b279 100644 --- a/unsloth/models/sentence_transformer.py +++ b/unsloth/models/sentence_transformer.py @@ -618,16 +618,23 @@ class FastSentenceTransformer(FastModel): transformers4 = Version(transformers.__version__).major < 5 model_type = "" try: - config = AutoConfig.from_pretrained(model_name, token=token, trust_remote_code=trust_remote_code) + config = AutoConfig.from_pretrained( + model_name, token = token, trust_remote_code = trust_remote_code + ) model_type = getattr(config, "model_type", "") except: pass - + is_distilbert = "distilbert" == model_type.lower() is_modernbert = "modernbert" == model_type.lower() is_debertav2 = "deberta-v2" == model_type.lower() - if "add_pooling_layer" not in kwargs and not is_distilbert and not is_modernbert and not is_debertav2: + if ( + "add_pooling_layer" not in kwargs + and not is_distilbert + and not is_modernbert + and not is_debertav2 + ): kwargs["add_pooling_layer"] = False # forces fp8 to be False since it's not supported