[BUG] Running import zeta immediately fails with "RuntimeError: mat1 and mat2 shapes cannot be multiplied (512x4 and 512x512)"

[julianantolin]

Hi,

I tried running vision mamba, which relies on zetascale. After following the error stack, I narrowed it down to zetascale causing the problem when trying to import it.

To Reproduce Steps to reproduce the behavior:

1. Try running 'import zeta' in Python

Expected behavior For zetascale to be imported and usable

Screenshots

Additional context I am working on Ubuntu 22.04, running Python 3.10.13 (I have had the same error with Python 3.9 and 3.11). I have the latest PyTorch as of February 21, 2024. I am working in a Conda virtual environment where I installed vision mamba by running 'pip install vision-mamba' which ran without issues.

I would greatly appreciate any advice on how to zetascale to work!

Thanks, Julian

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.

[github-actions[bot]]

Hello there, thank you for opening an Issue ! 🙏🏻 The team was notified and they will get back to you asap.

[kyegomez]

@julian540 can you upgrade your zeta?

pip3 install -U zetascale

[kyegomez]

@julian540 I'm closing for now, lmk if you have the issue