Update vision.py

2025-05-15 01:54:05 -07:00 · 2025-05-15 01:54:05 -07:00 · ab31b9af62
commit ab31b9af62
parent 505c27cd4c
1 changed files with 1 additions and 0 deletions
--- 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):