From 8f7e0164dfcd16c82e431f1eb381ce28b98dc898 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Thu, 30 Oct 2025 05:54:26 -0700 Subject: [PATCH] Update vision.py --- unsloth/models/vision.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/unsloth/models/vision.py b/unsloth/models/vision.py index 9ec8d7d09b..d6a9228b81 100644 --- a/unsloth/models/vision.py +++ b/unsloth/models/vision.py @@ -511,10 +511,11 @@ class FastBaseModel: if full_finetuning: os.environ["UNSLOTH_ENABLE_FULL_FINETUNING"] = "1" if dtype == torch.bfloat16: - print( - f"Unsloth: Using bfloat16 full finetuning which cuts memory usage by 50%.\n" - f"To enable float32 training, use `float32_mixed_precision = True` during FastLanguageModel.from_pretrained" - ) + if float32_mixed_precision != True: + print( + f"Unsloth: Using bfloat16 full finetuning which cuts memory usage by 50%.\n" + f"To enable float32 training, use `float32_mixed_precision = True` during FastLanguageModel.from_pretrained" + ) else: print("Unsloth: Float16 full finetuning uses more memory since we upcast weights to float32.") else: