From ab31b9af62999dd9da0d955e6e2737c09201ef42 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Thu, 15 May 2025 01:54:05 -0700 Subject: [PATCH] Update vision.py --- unsloth/models/vision.py | 1 + 1 file changed, 1 insertion(+) diff --git a/unsloth/models/vision.py b/unsloth/models/vision.py index 9005150a50..1fe05bbf86 100644 --- a/unsloth/models/vision.py +++ b/unsloth/models/vision.py @@ -320,6 +320,7 @@ class FastBaseModel: dtype = torch.float32 bnb_compute_dtype = eval(bnb_compute_dtype) pass + print("bnb_compute_dtype", bnb_compute_dtype) # Stop SDPA for some archs like Pixtral / Mistral3 if not ("attn_implementation" in kwargs):