Converted files of the original model at
mamba2-130m
to HF transformers compatible formats.
Not affiliated with both the original authors or hf.
Usage
from transformers import AutoModelForCausalLM
tokenizer = AutoTokenizer.from_pretrained("AntonV/mamba2-130m-hf")
model = AutoModelForCausalLM.from_pretrained("AntonV/mamba2-130m-hf")
input_ids = tokenizer("Hey how are you doing?", return_tensors="pt")["input_ids"]
out = model.generate(input_ids, max_new_tokens=10)
print(tokenizer.batch_decode(out))
Citation
BibTeX:
@inproceedings{mamba2,
title={Transformers are {SSM}s: Generalized Models and Efficient Algorithms Through Structured State Space Duality},
author={Dao, Tri and Gu, Albert},
booktitle={International Conference on Machine Learning (ICML)},
year={2024}
}
Runs of AntonV mamba2-130m-hf on huggingface.co
11.8K
Total runs
-783
24-hour runs
-174
3-day runs
-653
7-day runs
-1.3K
30-day runs
More Information About mamba2-130m-hf huggingface.co Model
mamba2-130m-hf huggingface.co is an AI model on huggingface.co that provides mamba2-130m-hf's model effect (), which can be used instantly with this AntonV mamba2-130m-hf model. huggingface.co supports a free trial of the mamba2-130m-hf model, and also provides paid use of the mamba2-130m-hf. Support call mamba2-130m-hf model through api, including Node.js, Python, http.
mamba2-130m-hf huggingface.co is an online trial and call api platform, which integrates mamba2-130m-hf's modeling effects, including api services, and provides a free online trial of mamba2-130m-hf, you can try mamba2-130m-hf online for free by clicking the link below.
AntonV mamba2-130m-hf online free url in huggingface.co:
mamba2-130m-hf is an open source model from GitHub that offers a free installation service, and any user can find mamba2-130m-hf on GitHub to install. At the same time, huggingface.co provides the effect of mamba2-130m-hf install, users can directly use mamba2-130m-hf installed effect in huggingface.co for debugging and trial. It also supports api for free installation.