Sketch the minimal mamba example. (#1465)

* Sketch the minimal mamba example.

* Fix rustfmt.

* Forward pass for mamba.

* Finish the forward pass.

* Inference fixes.

* Bugfixes.

* More fixes.

* Add a readme.
This commit is contained in:
Laurent Mazare
2023-12-22 00:28:50 +01:00
committed by GitHub
parent f6408a3779
commit ceb78d3e28
4 changed files with 458 additions and 1 deletions

View File

@ -40,4 +40,3 @@ fn criterion_benchmark(c: &mut Criterion) {
criterion_group!(benches, criterion_benchmark);
criterion_main!(benches);