J4bb4wukis
/

mamba2_172m_wikipedia_en_shuffeld

Model card Files Files and versions

J4bb4wukis commited on Jan 31

Commit

66d4d59

·

verified ·

1 Parent(s): c4f8fc3

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -35,8 +35,8 @@ pip install mamba-ssm
 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer
-mamba2 = AutoModelForCausalLM.from_pretrained("J4bb4wukis/mamba2_wikipedia_en_shuffeld")
-tokenizer = AutoTokenizer.from_pretrained("J4bb4wukis/mamba2_wikipedia_en_shuffeld")
 prompts = "Angela Merkel is"
 inputs = tokenizer(prompts,return_tensors='pt').input_ids

 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer
+mamba2 = AutoModelForCausalLM.from_pretrained("J4bb4wukis/mamba2_127m_wikipedia_en_shuffeld")
+tokenizer = AutoTokenizer.from_pretrained("J4bb4wukis/mamba2_127m_wikipedia_en_shuffeld")
 prompts = "Angela Merkel is"
 inputs = tokenizer(prompts,return_tensors='pt').input_ids