Update vision.py

This commit is contained in:
Daniel Han 2025-03-13 06:24:37 -07:00
commit 5bb6090140

View file

@ -249,7 +249,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 = "sdpa", #[TODO] Pixtral for eg fails
**kwargs,
)
# Return old flag