Update README.md
Browse files
README.md
CHANGED
@@ -14,7 +14,7 @@ import numpy as np
|
|
14 |
|
15 |
model=MambaLMHeadModel.from_pretrained("JunxiongWang/MambaByte_Books", device='cuda', dtype=torch.float32)
|
16 |
|
17 |
-
text = "
|
18 |
text_byte = np.frombuffer(text.encode('utf-8'), dtype=np.uint8)
|
19 |
input_ids = torch.from_numpy(text_byte[None, :]).long().cuda()
|
20 |
|
|
|
14 |
|
15 |
model=MambaLMHeadModel.from_pretrained("JunxiongWang/MambaByte_Books", device='cuda', dtype=torch.float32)
|
16 |
|
17 |
+
text = "In fair Verona, where we lay our scene, From ancient grudge, break to new mutiny, Where civil blood makes civil hands unclean."
|
18 |
text_byte = np.frombuffer(text.encode('utf-8'), dtype=np.uint8)
|
19 |
input_ids = torch.from_numpy(text_byte[None, :]).long().cuda()
|
20 |
|