Update README.md
Browse files
README.md
CHANGED
|
@@ -19,7 +19,7 @@ model=MambaLMHeadModel.from_pretrained("JunxiongWang/MambaByte_Code", device='cu
|
|
| 19 |
|
| 20 |
text = "import torch"
|
| 21 |
text_byte = np.frombuffer(text.encode('utf-8'), dtype=np.uint8)
|
| 22 |
-
input_ids = torch.from_numpy(text_byte[None, :]).long().cuda()
|
| 23 |
|
| 24 |
sample = model.generate(
|
| 25 |
input_ids=input_ids,
|
|
|
|
| 19 |
|
| 20 |
text = "import torch"
|
| 21 |
text_byte = np.frombuffer(text.encode('utf-8'), dtype=np.uint8)
|
| 22 |
+
input_ids = torch.from_numpy(text_byte[None, :].copy()).long().cuda()
|
| 23 |
|
| 24 |
sample = model.generate(
|
| 25 |
input_ids=input_ids,
|