unity
/

inference-engine-tiny-stories

Text Generation

unity-inference-engine

Model card Files Files and versions

UnityPaul commited on Apr 9, 2024

Commit

292a662

·

verified ·

1 Parent(s): e05e85c

Update RunTinyStories.cs

Files changed (1) hide show

RunTinyStories.cs +1 -1

RunTinyStories.cs CHANGED Viewed

@@ -80,7 +80,7 @@ public class RunTinyStories : MonoBehaviour
                 var row = FF.Select(model1.Forward(input)[8], 1, currentToken);
                 return FF.Multinomial(predictability * row, 1);
             },
-            (InputDef.FromModel(model1)[0], new InputDef(DataType.Int, new TensorShape()))
         );
         engine = WorkerFactory.CreateWorker(backend, model2);

                 var row = FF.Select(model1.Forward(input)[8], 1, currentToken);
                 return FF.Multinomial(predictability * row, 1);
             },
+            (model1.inputs[0], InputDef.Int(new TensorShape()))
         );
         engine = WorkerFactory.CreateWorker(backend, model2);