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

[pre-commit.ci] auto fixes from pre-commit.com hooks

Browse source 
for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2025-12-27 11:51:43 +00:00
commit 7913a68e8b
1 changed files with 6 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

8
unsloth/models/sentence_transformer.py
View file

@ -630,7 +630,9 @@ class FastSentenceTransformer(FastModel):
# if for_inference == True, skip Unsloth optimizations to avoid torch compile issues
if for_inference:
st_device = device_map
if isinstance(st_device, dict) or (isinstance(st_device, str) and st_device in ["auto", "sequential"]):
if isinstance(st_device, dict) or (
isinstance(st_device, str) and st_device in ["auto", "sequential"]
):
st_device = None
st_model = SentenceTransformer(
@ -742,7 +744,9 @@ class FastSentenceTransformer(FastModel):
)
st_device = device_map
if isinstance(st_device, dict) or (isinstance(st_device, str) and st_device in ["auto", "sequential"]):
if isinstance(st_device, dict) or (
isinstance(st_device, str) and st_device in ["auto", "sequential"]
):
st_device = None
st_model = SentenceTransformer(modules = modules, device = st_device)