Update vision.py

2025-03-13 06:25:18 -07:00 · 2025-03-13 06:25:18 -07:00 · 14b37cc5f4
commit 14b37cc5f4
parent 5bb6090140
1 changed files with 1 additions and 1 deletions
--- a/unsloth/models/vision.py
+++ b/unsloth/models/vision.py
@ -249,7 +249,7 @@ class FastBaseModel:
            # quantization_config   = bnb_config,
            token                   = token,
            trust_remote_code       = trust_remote_code,
-            # attn_implementation   = "sdpa", #[TODO] Pixtral for eg fails
+            attn_implementation     = "sdpa", #[TODO] Pixtral for eg fails
            **kwargs,
        )
        # Return old flag