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