whisper : set outputs from conv graph (#1959)

ggerganov · Mar 16, 2024 · 725350d · 725350d
1 parent 906c73b
commit 725350d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/whisper.cpp b/whisper.cpp
@@ -1680,6 +1680,8 @@ static struct ggml_cgraph * whisper_build_graph_conv(
  wstate.embd_enc = cur;
  }
 
+ ggml_set_output(cur);
+
  ggml_build_forward_expand(gf, cur);
 
  ggml_free(ctx0);