From: Georgi Gerganov <redacted>
Date: Sat, 16 Mar 2024 15:30:55 +0000 (+0200)
Subject: whisper : set outputs from conv graph (whisper/1959)
X-Git-Tag: upstream/0.0.1642~841
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=948380e93be514a9f826cec7332aa26401c73b8e;p=pkg%2Fggml%2Fsources%2Fggml

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

diff --git a/examples/whisper/whisper.cpp b/examples/whisper/whisper.cpp
index 289d7039..f6011974 100644
--- 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);