metadata

license: mit
library_name: unity-sentis
pipeline_tag: text-generation

Tiny Stories Model in Unity Sentis Format

This is the Tiny Stories model checked to run on Unity 2023. Tiny Stories is a Large Language Model that was trained on children's stories and can create stories based on the first couple of sentences.

How to Use

Create a new scene in Unity 2023
Install com.unity.sentis and com.unity.nuget.newtonsoft-json packages
Add the RunTinyStories.cs file to the Main Camera
Put tinystories.sentis, vocab.json and merges.txt in the Assets/StreamingAssets folder
Adjust some of the variables such as the outputText string to set the prompt
Press run
The output will appear in the console window

Example Input

One day an alien came down from Mars. It saw a chicken

Example Output

One day an alien came down from Mars. It saw a chicken and said, "Hello, little chicken. What are you doing here?"

The chicken replied, "I'm looking for a place to stay. I'm very tired."

The alien said, "You can stay here. I have a nice place for you. It's very comfortable."

The chicken was so happy. She thanked the alien and said, "Thank you. I'm very comfortable here."

The alien smiled and said, "You're welcome

Unity Sentis

Unity Sentis is the inference engine which runs on Unity 2023. More can be found about it here

Disclaimer

The model was trained on children's stories so very unlikely to produce undesirable text. As an extra precaution, we removed a few tokens from vocab.json that might not be suitable for younger audiences. The original json can be found on the Tiny Stories original page.