search

johnma2006 / mamba-minimal

Simple, minimal implementation of the Mamba SSM in one file of PyTorch.
Apache License 2.0
2.54k stars 188 forks source link

Seperate `class SSM` and `class MambaBlock`. #24

Open ChenLi2049 opened 6 months ago

ChenLi2049 commented 6 months ago

Seperate class SSM and class MambaBlock, so that def __init__ of both classes are clearer. I'm not sure if class SSM is supposed to be called class S6.