assafbk's picture
Update README.md
c4e3e56 verified
|
raw
history blame
854 Bytes
---
inference: false
license: mit
tags:
- text-generation
- mamba
- long context
---
# DeciMamba Checkpoint
The official checkpoint of DeciMamba-130m, finetuned for the Document Retrieval task as presented in [DeciMamba: Exploring the Length Extrapolation Potential of Mamba](https://arxiv.org/abs/2406.14528).
See our [Github Repo](https://github.com/assafbk/DeciMamba) for evalution and training scripts.
Bibtex:
```
@misc{benkish2024decimambaexploringlengthextrapolation,
title={DeciMamba: Exploring the Length Extrapolation Potential of Mamba},
author={Assaf Ben-Kish and Itamar Zimerman and Shady Abu-Hussein and Nadav Cohen and Amir Globerson and Lior Wolf and Raja Giryes},
year={2024},
eprint={2406.14528},
archivePrefix={arXiv},
primaryClass={cs.LG},
url={https://arxiv.org/abs/2406.14528},
}
```