JunxiongWang
/

MambaByte_Code

Text Generation

Inference Endpoints

Model card Files Files and versions Community

JunxiongWang commited on Sep 9

Commit

5690751

•

1 Parent(s): ea36284

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_Code", device='cu
 text = "import torch"
 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 = "import torch"
 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,