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