| | --- |
| | license: mit |
| | library_name: unity-sentis |
| | pipeline_tag: text-generation |
| | tags: |
| | - unity-inference-engine |
| | --- |
| | |
| | # Tiny Stories in Unity 6 with Inference Engine |
| |
|
| | This 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. |
| |
|
| | ## How to Use |
| |
|
| | * Create a new scene in Unity 6; |
| | * Install `com.unity.ai.inference` from the package manager; |
| | * Install `com.unity.nuget.newtonsoft-json` from the package manager; |
| | * Add the `RunTinyStories.cs` script to the Main Camera; |
| | * Drag the `tinystories.onnx` asset from the `models` folder into the `Model Asset` field; |
| | * Drag the `vocab.json` asset from the `data` folder into the `Vocab Asset` field; |
| | * Drag the `merges.txt` asset from the `data` folder into the `Merges Asset` field; |
| |
|
| | ## Preview |
| | Enter play mode. If working correctly the predicted text will be logged to the console. |
| |
|
| | ## Inference Engine |
| | Inference Engine is a neural network inference library for Unity. Find out more [here](https://docs.unity3d.com/Packages/com.unity.ai.inference@latest). |
| |
|
| | ## 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. |
