ModuleNotFoundError: No module named 'mamba_ssm.ops.triton.layernorm'

ChenhongyiYang / PlainMamba

[BMVC 2024] PlainMamba: Improving Non-hierarchical Mamba in Visual Recognition

Apache License 2.0

68 stars 7 forks source link

Closed LQchen1 closed 4 months ago

LQchen1 commented 4 months ago

wzy1845 commented 1 month ago

hi I also encountered the same problem. How did you solve it in the end? thank!

LQchen1 commented 1 month ago

Sorry man, it's been a while and I forgot, maybe you need to go to Mamba2's code for help.