OuteAI
/

Lite-Oute-2-Mamba2Attn-250M-Instruct

Model card Files Files and versions Community

edwko commited on 29 days ago

Commit

227e029

•

1 Parent(s): 0d376fc

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -198,7 +198,7 @@ The play "Romeo and Juliet" by William Shakespeare is a classic example of a tra
 ## Usage with HuggingFace transformers
 Model weights were converted from the original Mamba2 implementation to be Hugging Face compatible. <br>
 Due to the lack of official support for Mamba2 attention layers in Hugging Face Transformers, custom modeling files are included. <br>
-The attention layer implementation for the modeling files are based on the work from Pull Request #32027 in the Hugging Face Transformers repository: [https://github.com/huggingface/transformers/pull/32027](https://github.com/huggingface/transformers/pull/32027)
 To speed up inference, we recommend installing mamba-ssm and flash attention 2.

 ## Usage with HuggingFace transformers
 Model weights were converted from the original Mamba2 implementation to be Hugging Face compatible. <br>
 Due to the lack of official support for Mamba2 attention layers in Hugging Face Transformers, custom modeling files are included. <br>
+The implementation of Mamba2 with attention in the modeling files comes from Pull Request #32027 in the Hugging Face Transformers repository: [https://github.com/huggingface/transformers/pull/32027](https://github.com/huggingface/transformers/pull/32027)
 To speed up inference, we recommend installing mamba-ssm and flash attention 2.