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):