Update README.md

README.md

@@ -20,7 +20,7 @@ model=MambaLMHeadModel.from_pretrained("JunxiongWang/MambaByte_Stories", device=
 text = "It was terribly cold and nearly dark on the last evening of the old year, and the snow was falling fast."
 
 text_byte = np.frombuffer(text.encode('utf-8'), dtype=np.uint8)
-input_ids = torch.from_numpy(text_byte[None, :]).long().cuda()
+input_ids = torch.from_numpy(text_byte[None, :].copy()).long().cuda()
 
 sample = model.generate(
     input_ids=input_ids,