Hugging Face's logo

aditya-6122
/
Tiny-Stories-GRU-LanguageModel-ByteLevelEncoding

Text Generation
PyTorch
English
language-model
rnn
gru
tiny-stories
bpe-tokenizer
Model card Files
xet
 Community
aditya-6122 commited on
Commit
8f35ebb
·
verified ·
1 Parent(s): 0474187

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -1
README.md CHANGED
@@ -14,13 +14,16 @@ datasets:
14
  pipeline_tag: text-generation
15
  ---
16
 
17
- # vanilla-rnn-gru-like
18
 
19
  ## Model Details
20
 
21
  ### Model Description
22
  This is a custom language model trained on a dataset of short stories, designed for text generation tasks.
23
 
 
 
 
24
  ### Model Sources
25
  - **Repository**: [Aditya6122/BuildingLanguageModel-TinyStories](https://github.com/Aditya6122/BuildingLanguageModel-TinyStories)
26
 
 
14
  pipeline_tag: text-generation
15
  ---
16
 
17
+ # Tiny-Stories-GRU-LanguageModel-ByteLevelEncoding
18
 
19
  ## Model Details
20
 
21
  ### Model Description
22
  This is a custom language model trained on a dataset of short stories, designed for text generation tasks.
23
 
24
+ ### Architecture
25
+ <img src="https://huggingface.co/aditya-6122/Tiny-Stories-GRU-LanguageModel-ByteLevelEncoding/resolve/main/model_arch.jpg" alt="Architecture" width=850 />
26
+
27
  ### Model Sources
28
  - **Repository**: [Aditya6122/BuildingLanguageModel-TinyStories](https://github.com/Aditya6122/BuildingLanguageModel-TinyStories)
29