[BUG] Cannot import name 'SSM' from 'zeta.nn'

[MarioPaps]

Describe the bug The zeta module which is a required import does not have the attribute ssm

To Reproduce Steps to reproduce the behavior:

1. Go to 'vision_mamba/model.py'
2. Click on 'model.py'
3. If you run example.py in the root directory, you will get an import Error: ImportError: cannot import name 'SSM' from 'zeta.nn' (C:\Users\mseb\miniconda3\envs\py38\lib\site-packages\zeta\nn_init_.py)

Expected behavior We would expect the import to work

Additional context I also tried 'from zeta.nn.modules import SSM'. This also does not work although the documentation here [https://zeta.apac.ai/en/latest/zeta/nn/modules/ssm/] suggests that it should work.

Upvote & Fund

• We're using Polar.sh so you can upvote and help fund this issue.
• We receive the funding once the issue is completed & confirmed by you.
• Thank you in advance for helping prioritize & fund our backlog.