Downloads · 30 days
3.2K
8% of all-time downloads
mamba-130m
ArthurZ/mamba-130m
mamba-130m is a text generation model from ArthurZ. Use it when you need the model to write or continue text. It is set up for transformers.
All-time downloads
40.7K
Public
Parameters
129M
517 MB on disk
Likes
3
Public
Repo makeup
Click a slice to open those files.
.safetensors517 MB · 100%
At a glance
- Task
- Text Generation
- Library
- transformers
- Model type
- mamba
- Access
- Public
- Created
- Feb 16, 2024
- Updated
- Feb 19, 2024
- SHA
- 80bb8cfa
Try a prompt
Model card
From the Hugging Face model README
>>> from transformers import MambaConfig, MambaForCausalLM, AutoTokenizer
>>> import torch
>>> tokenizer = AutoTokenizer.from_pretrained("EleutherAI/gpt-neox-20b", padding_side = "left")
>>> tokenizer.pad_token = tokenizer.eos_token
>>> model = MambaForCausalLM.from_pretrained("state-spaces/mamba-130m", vocab_size=50280, num_hidden_layers=24, torch_dtype=torch.float32)
>>> model.config.use_cache = True
>>> input_ids = tokenizer(["Hey how are you doing?", "Explain how soy sauce is made"], padding=True, return_tensors= "pt")["input_ids"]
>>> out = model.generate(input_ids, max_new_tokens=10)
>>> print(tokenizer.batch_decode(out))
["<|endoftext|>Hey how are you doing?\n\nI'm a newbie to the game", 'Explain how soy sauce is made.\n\n1. Add the soy sauce to']
- Task
- Text Generation
- Library
- transformers
- Type
- mamba
- Created
- Feb 16, 2024
- Updated
- Feb 19, 2024