YenChunChen's picture
make flash attention usage configurable from user code
bfc0c05