Hugging Face's logo

DdIiVvYyAaMm
/
mamba-370m-story-generation

Text Generation
Transformers
Safetensors
English
mamba
Story-Generation
State-Space
text-generation-inference
story-writing
Inference Endpoints
Model card Files Community
DdIiVvYyAaMm commited on
Commit
f6cde40
·
verified ·
1 Parent(s): 13011c0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -36,8 +36,8 @@ Perplexity : 207
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
  ```
 
36
  >>> from transformers import MambaForCausalLM, AutoTokenizer
37
  >>> model_name = "DdIiVvYyAaMm/mamba-370m-story-generation"
38
 
39
+ >>> tokenizer = AutoTokenizer.from_pretrained(model_name)
40
+ >>> model = MambaForCausalLM.from_pretrained(model_name)
41
 
42
  # And the rest of code standard as from transformers library.
43
  ```