Update RunTinyStories.cs
Browse files- RunTinyStories.cs +1 -1
RunTinyStories.cs
CHANGED
|
@@ -80,7 +80,7 @@ public class RunTinyStories : MonoBehaviour
|
|
| 80 |
var row = FF.Select(model1.Forward(input)[8], 1, currentToken);
|
| 81 |
return FF.Multinomial(predictability * row, 1);
|
| 82 |
},
|
| 83 |
-
(
|
| 84 |
);
|
| 85 |
|
| 86 |
engine = WorkerFactory.CreateWorker(backend, model2);
|
|
|
|
| 80 |
var row = FF.Select(model1.Forward(input)[8], 1, currentToken);
|
| 81 |
return FF.Multinomial(predictability * row, 1);
|
| 82 |
},
|
| 83 |
+
(model1.inputs[0], InputDef.Int(new TensorShape()))
|
| 84 |
);
|
| 85 |
|
| 86 |
engine = WorkerFactory.CreateWorker(backend, model2);
|