mirror of
https://github.com/huggingface/candle.git
synced 2025-06-18 19:47:12 +00:00
1e26d539d9
* Sketch the mamba model for inference. * Complete the forward pass. * Add the mamba example. * Optimize the selective-scan part. * Fix a couple shape mismatches and get inference to work. * Tweak the readmes. * More readme tweaks.
16 lines
854 B
Markdown
16 lines
854 B
Markdown
# candle-mamba-minimal: minimal implementation of Mamba
|
|
|
|
This is based on [mamba-minimal](https://github.com/johnma2006/mamba-minimal).
|
|
|
|
Compared to the mamba example, this version can handle training but is much
|
|
slower.
|
|
|
|
## Running the example
|
|
|
|
```bash
|
|
$ cargo run --example mamba-minimal --release -- --prompt "Mamba is the"
|
|
Mamba is the most popular and best-selling game in the world. It has been downloaded more than 1,000 times by over 1 million people worldwide since its release on March 18th 2016.
|
|
|
|
The Mamba series of games are a collection that combines elements from all genres including action, adventure, strategy & puzzle games with some unique gameplay features such as stealth and survival. The game is also known for its innovative graphics and the ability to play in a variety of different modes like single player or multiplayer.
|
|
```
|