search

tommyip / mamba2-minimal

Minimal Mamba-2 implementation in PyTorch
Apache License 2.0
129 stars 13 forks source link

AssertionError in ssd function #1

Open Yiwen233 opened 4 months ago

Yiwen233 commented 4 months ago

When I try to modify your released simple mamba2 code in my experiment I got this assertion error. I wonder why should make this assumption and the official code does not have this constraint.

File "/home/zyw/workspace/mamba-minimal", line 232, in ssd
    assert x.shape[1] % chunk_size == 0
AssertionError
radarFudan commented 2 months ago

I guess this constraint come from the minimal version in mamba

https://github.com/state-spaces/mamba/blob/219f03c840d5a44e7d42e4e728134834fddccf45/mamba_ssm/modules/ssd_minimal.py#L45