Update vision.py

This commit is contained in:
Daniel Han 2025-03-12 22:22:23 -07:00
commit 2079f9f217

View file

@ -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