From 5bb60901407b0d50ae8777fa28d1bd19e996cf79 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Thu, 13 Mar 2025 06:24:37 -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 fbb154ddd0..8436791061 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 = "eager", #[TODO] Pixtral for eg fails + # attn_implementation = "sdpa", #[TODO] Pixtral for eg fails **kwargs, ) # Return old flag