Laurent Mazare b3ea96b62b Add a prompt and support more models in llama2-c. (#285) ...

* Support more models in llama2-c.

* Add a prompt.

2023-07-31 13:09:30 +01:00

..

examples

Add a prompt and support more models in llama2-c. (#285)

2023-07-31 13:09:30 +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 a flag to save the trained weights. (#279)

2023-07-30 15:41:42 +01:00