How to use Zyphra/Mamba-370M with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("Zyphra/Mamba-370M", dtype="auto")