JunxiongWang
/

MambaByte_PG19_972M

Text Generation

Inference Endpoints

Model card Files Files and versions Community

JunxiongWang commited on Sep 9

Commit

3bd6751

•

1 Parent(s): 2f582fc

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -27,7 +27,7 @@ model=MambaLMHeadModel.from_pretrained("JunxiongWang/MambaByte_PG19_972M", devic
 text = "It is a truth universally acknowledged, that a single man in possession of a good fortune, must be in want of a wife,"
 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 = "It is a truth universally acknowledged, that a single man in possession of a good fortune, must be in want of a wife,"
 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,