Skip to content

Commit

Permalink
wip
Browse files Browse the repository at this point in the history
  • Loading branch information
eegli committed Feb 3, 2025
1 parent 8498712 commit 0eff338
Showing 1 changed file with 12 additions and 0 deletions.
12 changes: 12 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -30,12 +30,24 @@ You will need to **install a recent PyTorch version manually**. We use `>=2.6.0`
pip install 'torch>=2.6.0' --index-url https://download.pytorch.org/whl/cu124
```

For `uv`:

```
uv pip install 'torch>=2.6.0' --index-url https://download.pytorch.org/whl/cu124
```

Finally, in order to use the efficient [Mamba-SSM](https://github.com/state-spaces/mamba), follow their instructions on the homepage. You'll need Linux and a GPU available during installation.

```
pip install mamba-ssm>=2.2.2 causal-conv1d>=1.4.0 --no-build-isolation
```

For `uv`:

```
uv pip install mamba-ssm>=2.2.2 causal-conv1d>=1.4.0 --no-build-isolation
```

If `mamba-ssm` is not available, we fall back to using `mambapy`, which is written in pure PyTorch.

## Quickstart
Expand Down

0 comments on commit 0eff338

Please sign in to comment.