Converted models fitting working with https://github.com/vasqu/mamba2-torch