If I have an input of [batchsize,length,dim], how can I use the model

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

Open renyuanzhe opened 6 months ago