Update README.md

README.md

@@ -35,11 +35,11 @@ pip install mamba-ssm
 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-xlstm = AutoModelForCausalLM.from_pretrained("J4bb4wukis/mamba2_wikipedia_en_shuffeld")
+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
-outputs = xlstm.generate(inputs, max_new_tokens=100, do_sample=True, top_k=10, top_p=0.95)
+outputs = mamba2.generate(inputs, max_new_tokens=100, do_sample=True, top_k=10, top_p=0.95)
 print(tokenizer.batch_decode(outputs, skip_special_tokens=True))
 ```