huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 10:38:54 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
fa2b64d678ca83e2fbc3dabdecffbc778d5b067d
candle/candle-flash-attn
History
Laurent Mazare fa2b64d678 Proper flash-attn parameters. (#244) 
* Proper flash-attn parameters.

* Set the flash attention parameters.

* Add more validations.

* Setup the o_ flash attn parameters.

* More flash-attn support.

* Set more flash attn parameters.
2023-07-26 10:13:40 +01:00
..
cutlass @ c4f6b8c6bc
Add flash attention (#241)
2023-07-26 07:48:10 +01:00
kernels
Proper flash-attn parameters. (#244)
2023-07-26 10:13:40 +01:00
src
Proper flash-attn parameters. (#244)
2023-07-26 10:13:40 +01:00
build.rs
Specific cache dir for the flash attn build artifacts. (#242)
2023-07-26 08:04:02 +01:00
Cargo.toml
Add flash attention (#241)
2023-07-26 07:48:10 +01:00