From 64136e6336c6db755f790624b496493b36685ccf Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Thu, 30 Oct 2025 06:27:58 -0700 Subject: [PATCH] Update vision.py --- unsloth/models/vision.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/unsloth/models/vision.py b/unsloth/models/vision.py index 4947bae64c..e897145acc 100644 --- a/unsloth/models/vision.py +++ b/unsloth/models/vision.py @@ -519,13 +519,13 @@ class FastBaseModel: 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" ) - os.environ["UNSLOTH_BFLOAT16_MIXED_PRECISION"] = "1" else: print( f"Unsloth: Using full float32 full finetuning.\n" f"To enable bfloat16 training to reduce VRAM usage by 50% albeit with a slightly higher loss, "\ "use `float32_mixed_precision = False` during FastLanguageModel.from_pretrained" ) + os.environ["UNSLOTH_BFLOAT16_MIXED_PRECISION"] = "1" else: print("Unsloth: Float16 full finetuning uses more memory since we upcast weights to float32.") else: