diff --git a/unsloth/models/vision.py b/unsloth/models/vision.py index f4be016770..c6fafe1e4b 100644 --- a/unsloth/models/vision.py +++ b/unsloth/models/vision.py @@ -234,7 +234,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 = "eager", [TODO] Pixtral for eg fails **kwargs, ) # Return old flag