diff --git a/unsloth/models/sentence_transformer.py b/unsloth/models/sentence_transformer.py index e93a2a7978..e57ada7807 100644 --- a/unsloth/models/sentence_transformer.py +++ b/unsloth/models/sentence_transformer.py @@ -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