ggerganov commited on
Commit
62505d4
·
unverified ·
1 Parent(s): 0c378f2

whisper : set outputs from conv graph (#1959)

Browse files
Files changed (1) hide show
  1. whisper.cpp +2 -0
whisper.cpp CHANGED
@@ -1680,6 +1680,8 @@ static struct ggml_cgraph * whisper_build_graph_conv(
1680
  wstate.embd_enc = cur;
1681
  }
1682
 
 
 
1683
  ggml_build_forward_expand(gf, cur);
1684
 
1685
  ggml_free(ctx0);
 
1680
  wstate.embd_enc = cur;
1681
  }
1682
 
1683
+ ggml_set_output(cur);
1684
+
1685
  ggml_build_forward_expand(gf, cur);
1686
 
1687
  ggml_free(ctx0);