roneneldan commited on
Commit
190d22e
1 Parent(s): 33518ee

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -0
README.md CHANGED
@@ -4,6 +4,10 @@ datasets:
4
  ---
5
  Model trained on the TinyStories Dataset, see https://arxiv.org/abs/2305.07759
6
 
 
 
 
 
7
  ------ EXAMPLE USAGE ---
8
 
9
  from transformers import AutoModelForCausalLM, AutoTokenizer, GenerationConfig
 
4
  ---
5
  Model trained on the TinyStories Dataset, see https://arxiv.org/abs/2305.07759
6
 
7
+ Based on GPT-Neo architecture.
8
+
9
+ License: mit
10
+
11
  ------ EXAMPLE USAGE ---
12
 
13
  from transformers import AutoModelForCausalLM, AutoTokenizer, GenerationConfig