Update README.md

README.md

@@ -19,7 +19,7 @@ model=MambaLMHeadModel.from_pretrained("JunxiongWang/MambaByte_Books", device='c
 
 text = "In fair Verona, where we lay our scene, From ancient grudge, break to new mutiny, Where civil blood makes civil hands unclean."
 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,