ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

refactor do_lower_case

Browse source
This commit is contained in:
electroglyph 2025-12-16 02:01:31 -08:00
commit a099456372
1 changed files with 2 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

6
unsloth/models/sentence_transformer.py
View file

@ -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) | {