Hugging Face's logo

roneneldan
/
TinyStories-33M

Text Generation
Transformers
PyTorch
gpt_neo
Inference Endpoints
Model card Files Community
18
roneneldan commited on
Commit
a139e23
•
1 Parent(s): aa90ce0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -4,7 +4,7 @@ Model trained on the TinyStories Dataset, see https://arxiv.org/abs/2305.07759
4
 
5
  from transformers import AutoModelForCausalLM, AutoTokenizer, GenerationConfig
6
 
7
- model = AutoModelForCausalLM.from_pretrained('roneneldan/TinyStories-1M')
8
 
9
  tokenizer = AutoTokenizer.from_pretrained("EleutherAI/gpt-neo-125M")
10
 
 
4
 
5
  from transformers import AutoModelForCausalLM, AutoTokenizer, GenerationConfig
6
 
7
+ model = AutoModelForCausalLM.from_pretrained('roneneldan/TinyStories-33M')
8
 
9
  tokenizer = AutoTokenizer.from_pretrained("EleutherAI/gpt-neo-125M")
10