unity
/

sentis-tiny-stories

Text Generation

Model card Files Files and versions Community

UnityPaul commited on Apr 9

Commit

292a662

•

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);