From 2079f9f21762d2f5840ca52279dab4aea061a9fe Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Wed, 12 Mar 2025 22:22:23 -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 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