transformers icon indicating copy to clipboard operation
transformers copied to clipboard

Add siglip flashattention support?

Open lucasjinreal opened this issue 1 year ago • 5 comments

Add siglip flashattention support?

I noticed that the navit-flashatten-siglip version already added flash attention, would transformers add it support inside ?

lucasjinreal avatar May 30 '24 11:05 lucasjinreal

cc @molbap or @qubvel maybe :)

LysandreJik avatar May 31 '24 10:05 LysandreJik

these two guys looked but didn't response.

lucasjinreal avatar May 31 '24 14:05 lucasjinreal

@lucasjinreal thanks for the feature request, I will take a look at it!

qubvel avatar May 31 '24 15:05 qubvel

thanks, since more and more MLLM used siglip as vision encoder, if flash attention support would be very reduce training cost time

lucasjinreal avatar May 31 '24 16:05 lucasjinreal