whisper : set outputs from conv graph (whisper/1959)

diff --git a/examples/whisper/whisper.cpp b/examples/whisper/whisper.cpp
index 289d70398493a91602636995486297813ea0e1e4..f60119749629d82f0b836e0d939ed44930ebbf34 100644 (file)
--- a/examples/whisper/whisper.cpp
+++ b/examples/whisper/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);