Update README.md
Browse files
README.md
CHANGED
@@ -12,6 +12,8 @@ model = MambaLMHeadModel.from_pretrained("Zyphra/Mamba-370M", iteration=10_000,
|
|
12 |
|
13 |
If iteration is not specified, then the model from the root of the repository is loaded, which is the final iteration (610,351).
|
14 |
|
|
|
|
|
15 |
Here is a snippet for text generation:
|
16 |
```
|
17 |
import transformers, torch
|
|
|
12 |
|
13 |
If iteration is not specified, then the model from the root of the repository is loaded, which is the final iteration (610,351).
|
14 |
|
15 |
+
The model was trained using `"EleutherAI/gpt-neox-20b"` tokenizer.
|
16 |
+
|
17 |
Here is a snippet for text generation:
|
18 |
```
|
19 |
import transformers, torch
|