Team Ai
Modelpublic

unity/inference-engine-tiny-stories

sourceHugging Facemitupdated 4mo agoView on Hugging Face
12likes54downloads
README.md29 linesDownload Raw Back to root
1---2license: mit3library_name: unity-sentis4pipeline_tag: text-generation5tags:6  - unity-inference-engine7---8 9# Tiny Stories in Unity 6 with Inference Engine10 11This is the [Tiny Stories model](https://huggingface.co/roneneldan/TinyStories-33M) running in Unity 6 with Inference Engine. 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.12 13## How to Use14 15* Create a new scene in Unity 6;16* Install `com.unity.ai.inference` from the package manager;17* Add the `RunTinyStories.cs` script to the Main Camera;18* Drag the `tinystories.onnx` asset from the `models` folder into the `Model Asset` field;19* Drag the `vocab.json` asset from the `data` folder into the `Vocab Asset` field;20* Drag the `merges.txt` asset from the `data` folder into the `Merges Asset` field;21 22## Preview23Enter play mode. If working correctly the predicted text will be logged to the console.24 25## Inference Engine26Inference Engine is a neural network inference library for Unity. Find out more [here](https://docs.unity3d.com/Packages/com.unity.ai.inference@latest).27 28## Disclaimer29The 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.