├── .gitignore ├── LICENSE ├── README.md ├── demo.ipynb ├── mamba2.py └── requirements.txt /.gitignore: -------------------------------------------------------------------------------- 1 | .venv/ 2 | __pycache__/ 3 | .ipynb_checkpoints/ 4 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tommyip/mamba2-minimal/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tommyip/mamba2-minimal/HEAD/README.md -------------------------------------------------------------------------------- /demo.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tommyip/mamba2-minimal/HEAD/demo.ipynb -------------------------------------------------------------------------------- /mamba2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tommyip/mamba2-minimal/HEAD/mamba2.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | torch 2 | transformers 3 | einops 4 | --------------------------------------------------------------------------------