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

..

convert_checkpoint.py

Resurrect the llama npy support. (#140)

2023-07-11 19:32:10 +01:00

main.rs

Better handling of dtypes in llama. (#243)

2023-07-26 08:28:33 +01:00

model.rs

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

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