From a0994563728cfd9f2585d53bd4a7e2bc9bc70765 Mon Sep 17 00:00:00 2001 From: electroglyph Date: Tue, 16 Dec 2025 02:01:31 -0800 Subject: [PATCH] refactor do_lower_case --- unsloth/models/sentence_transformer.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/unsloth/models/sentence_transformer.py b/unsloth/models/sentence_transformer.py index 3300b16928..838563a448 100644 --- a/unsloth/models/sentence_transformer.py +++ b/unsloth/models/sentence_transformer.py @@ -67,7 +67,7 @@ class FastSentenceTransformer(FastModel): old_environ = os.environ.get("UNSLOTH_WARN_UNINITIALIZED", "1") os.environ["UNSLOTH_WARN_UNINITIALIZED"] = "0" - + try: model, tokenizer = FastModel.from_pretrained( model_name = model_name, @@ -103,9 +103,7 @@ class FastSentenceTransformer(FastModel): torch.nn.Module.__init__(transformer_module) transformer_module.auto_model = model transformer_module.tokenizer = tokenizer - transformer_module.do_lower_case = False - if hasattr(tokenizer, "do_lower_case"): - transformer_module.do_lower_case = tokenizer.do_lower_case + transformer_module.do_lower_case = getattr(tokenizer, "do_lower_case", False) model_forward_params = list(inspect.signature(model.forward).parameters) transformer_module.model_forward_params = set(model_forward_params) | {