DdIiVvYyAaMm
commited on
Update README.md
Browse files
README.md
CHANGED
@@ -35,9 +35,9 @@ Perplexity : 207
|
|
35 |
```python
|
36 |
>>> from transformers import MambaForCausalLM, AutoTokenizer
|
37 |
>>> model_name = "DdIiVvYyAaMm/mamba-370m-story-generation"
|
38 |
-
|
39 |
>>> tokenizer = AutoTokenizer.from_pretrained(chk_path)
|
40 |
>>> model = MambaForCausalLM.from_pretrained(chk_path)
|
41 |
-
|
42 |
-
|
43 |
```
|
|
|
35 |
```python
|
36 |
>>> from transformers import MambaForCausalLM, AutoTokenizer
|
37 |
>>> model_name = "DdIiVvYyAaMm/mamba-370m-story-generation"
|
38 |
+
|
39 |
>>> tokenizer = AutoTokenizer.from_pretrained(chk_path)
|
40 |
>>> model = MambaForCausalLM.from_pretrained(chk_path)
|
41 |
+
|
42 |
+
# And the rest of code standard as from transformers library.
|
43 |
```
|