Support for vision models

[zhxieml]

Thanks for the great work!

I was wondering if non-causal attention will be supported in the near future to accommodate VLMs. Currently, only causal attention is available.

ring-flash-attention/ring_flash_attn/adapters/hf_adapter.py

Line 142 in 7866779

assert causal, "only causal attention is supported yet."