Rami Cassia

Hi all, I am having the same problem. I would like to use Mamba2. I tried the following as suggested: pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cu118 pip install causal_conv1d==1.1.1...

this is what worked for me in the end: !pip install torch==2.4.0 torchvision==0.19.0 torchaudio==2.4.0 --index-url https://download.pytorch.org/whl/cu121 !pip uninstall mamba-ssm causal-conv1d !pip install causal-conv1d && pip install mamba-ssm Mamba2 runs fine...

> > this is what worked for me in the end: > > > > > > !pip install torch==2.4.0 torchvision==0.19.0 torchaudio==2.4.0 --index-url https://download.pytorch.org/whl/cu121 !pip uninstall mamba-ssm causal-conv1d !pip install...