Fix tests and drift from main

This commit is contained in:
Jesse Gross
2025-03-07 16:05:56 -08:00
committed by Michael Yang
parent c62861f4fa
commit 0e886595bf
3 changed files with 6 additions and 2 deletions

View File

@@ -179,7 +179,7 @@ func (m *Model) Forward(ctx ml.Context, opts input.Options) (ml.Tensor, error) {
return nil, err
}
outputs, err := ctx.FromIntSlice(opts.Outputs, len(opts.Outputs))
outputs, err := ctx.Output().FromIntSlice(opts.Outputs, len(opts.Outputs))
if err != nil {
return nil, err
}