From 14b37cc5f473bf0bc6f95cea9a53ac0cfeba99d8 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Thu, 13 Mar 2025 06:25:18 -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 8436791061..2ef9d2ee99 100644 --- 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