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

for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2026-01-05 10:23:35 +00:00
commit 0c2425cb8d

View file

@ -778,7 +778,9 @@ class FastSentenceTransformer(FastModel):
else:
dtype = torch.float32
elif dtype == torch.bfloat16 and not SUPPORTS_BFLOAT16:
print("Unsloth: Device does not support bfloat16. Using float16 instead.")
print(
"Unsloth: Device does not support bfloat16. Using float16 instead."
)
dtype = torch.float16
# Determine device