Update vision.py

This commit is contained in:
Daniel Han 2025-03-13 06:39:17 -07:00
commit cb0d41e30e

View file

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