search

xi-j / Mamba-TasNet

GNU General Public License v3.0
50 stars 3 forks source link

No module named 'mamba_ssm.ops.triton.layernorm' #1

Closed 6eternal6 closed 2 months ago

6eternal6 commented 4 months ago

"from mamba_ssm.ops.triton.layernorm import RMSNorm, layer_norm_fn, rms_norm_fn" have error，when use try except to import, such as "try: from mamba_ssm.ops.triton.layernorm import RMSNorm, layer_norm_fn, rms_norm_fn except ImportError: RMSNorm, layer_norm_fn, rms_norm_fn = None, None, None"
the python console does not return an error, but RMSNorm, layer_norm_fn, rms_norm_fn = None, None, None

We looked through dir(mamba_ssm.ops.triton) , but layernorm does exist.

xi-j commented 2 months ago

Make sure the version of mamba_ssm is compatible to your hardware and CUDA. The below packages & versions work for us (see requirements.txt): causal-conv1d==1.1.3.post1 mamba-ssm==1.1.3.post1 torch==2.0.1 --index-url https://download.pytorch.org/whl/cu118