From 8dd99de4eec1f32773558c0401f152376d02fd4e Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Wed, 12 Mar 2025 22:23:29 -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 c6fafe1e4b..356aa10dcb 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 = "eager", [TODO] Pixtral for eg fails + attn_implementation = "eager", #[TODO] Pixtral for eg fails **kwargs, ) # Return old flag