search

SkyworkAI / Gamba

Gamba: Marry Gaussian Splatting with Mamba for single view 3D reconstruction
https://florinshen.github.io/gamba-project
90 stars 4 forks source link

which version of mamba_ssm should I install？ #3

Closed 1having0 closed 2 months ago

1having0 commented 3 months ago

from mamba_ssm.ops.triton.layernorm import RMSNorm, layer_norm_fn, rms_norm_fn ModuleNotFoundError: No module named 'mamba_ssm.ops.triton.layernorm'

florinshen commented 2 months ago

The mamba-ssm package used in this project is v1.2.2, you may check this download link of pre-compiled wheels. https://github.com/state-spaces/mamba/releases/download/v1.2.2/mamba_ssm-1.2.2+cu118torch2.1cxx11abiTRUE-cp310-cp310-linux_x86_64.whl