Laurent Mazare 3e89df938c Starcoder fix (#264) ...

* Bugfix for starcoder.

* Get some proper code generation.

* Slightly simpler softmax.

2023-07-28 11:17:49 +01:00

..

examples

Starcoder fix (#264)

2023-07-28 11:17:49 +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

TP sharding v2

2023-07-27 09:58:14 +02:00