JunxiongWang
/

MambaByte_Arxiv

Text Generation

Inference Endpoints

Model card Files Files and versions Community

JunxiongWang commited on Sep 9

Commit

1b4d270

•

1 Parent(s): 784a648

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -19,7 +19,7 @@ model=MambaLMHeadModel.from_pretrained("JunxiongWang/MambaByte_Arxiv", device='c
 text = "\documentclass[12pt]{article}"
 text_byte = np.frombuffer(text.encode('utf-8'), dtype=np.uint8)
-input_ids = torch.from_numpy(text_byte[None, :]).long().cuda()
 sample = model.generate(
     input_ids=input_ids,

 text = "\documentclass[12pt]{article}"
 text_byte = np.frombuffer(text.encode('utf-8'), dtype=np.uint8)
+input_ids = torch.from_numpy(text_byte[None, :].copy()).long().cuda()
 sample = model.generate(
     input_ids=input_ids,