Views
No views yet
from model import Mamba, ModelArgs
mamba_model = Mamba.from_pretrained("pt-sk/mamba").to("cuda")
tokenizer = AutoTokenizer.from_pretrained('pt-sk/mamba')