From 725350d4ea1545d890fe41f815b851cbc57838f6 Mon Sep 17 00:00:00 2001
From: Georgi Gerganov <ggerganov@gmail.com>
Date: Sat, 16 Mar 2024 17:30:55 +0200
Subject: [PATCH] whisper : set outputs from conv graph (#1959)

---
 whisper.cpp | 2 ++
 1 file changed, 2 insertions(+)

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