[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
This commit is contained in:
parent
65fc139032
commit
0c2425cb8d
1 changed files with 3 additions and 1 deletions
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue