ottonemo / ted

Text Generation with TED - Trainable Exponential Decay(s)

Text Generation with TED - Trainable Exponential Decay(s)

Go to the project's GitHub Page to learn more.

Project structure

• train.py: Trains the Tiny Stories TED model.
• gen.py: Generates the text using trained model.
• index.js: Inference code for TED architecture in JavaScript.
• index.html: The project's GitHub Page source code
• tiny_stories/: The weights and config files for Tiny Stories TED model.
• ted/: The source code for TED architecture.
• assets/: The resources used for project's GitHub Page (images).