yury-zyphra commited on
Commit
e39a3c9
·
verified ·
1 Parent(s): ba3dce0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -0
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