Laurent Mazare f052ba76cb Lining up the flash attn version with the non-flash one. (#248) ...

* Move the flash-attn function in the proper crate.

* Causality tweak.

2023-07-26 15:11:45 +01:00

..

examples

Lining up the flash attn version with the non-flash one. (#248)

2023-07-26 15:11:45 +01:00

src

Add some 'cuda-if-available' helper function. (#172)

2023-07-15 08:25:15 +01:00

build.rs

Add flash attention (#241)

2023-07-26 07:48:10 +01:00

Cargo.toml

Add flash attention (#241)

2023-07-26 07:48:10 +01:00