huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-20 12:06:35 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
3f13ad3d79790d02136f4e3ebcd2c8b671615a23
candle/candle-flash-attn/src
History
Laurent Mazare 7ebc3548e1 Use flash-attn in gemma. (#2195) 
* Use flash-attn in gemma.

* Fix flash-attn for head dim 256.
2024-05-18 19:18:59 +02:00
..
ffi.rs
chore: update flash attention kernels (#1518)
2024-01-05 18:28:55 +01:00
lib.rs
Use flash-attn in gemma. (#2195)
2024-05-18 19:18:59 +02:00