[BUG] bug

[qianxule]

Traceback (most recent call last):
File "/home/liguanting/project/AdaIR/net/mambaformer.py", line 24, in
out = model(x, return_loss=True) # Perform a forward pass and calculate the loss
File "/home/liguanting/anaconda3/envs/adairvim/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1511, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/home/liguanting/anaconda3/envs/adairvim/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1520, in _call_impl
return forward_call(*args, **kwargs)
File "/home/liguanting/anaconda3/envs/adairvim/lib/python3.10/site-packages/mamba_former/main.py", line 126, in forward
x = self.input_mamba(x) + skip
File "/home/liguanting/anaconda3/envs/adairvim/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1511, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/home/liguanting/anaconda3/envs/adairvim/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1520, in _call_impl
return forward_call(*args, **kwargs)
File "/home/liguanting/anaconda3/envs/adairvim/lib/python3.10/site-packages/zeta/nn/modules/simple_mamba.py", line 107, in forward
(b, l, d) = x.shape
ValueError: not enough values to unpack (expected 3, got 2)

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]

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