whisper : set outputs from conv graph (#1959)

2026-10-04 13:31:18 +02:00 · 2024-03-16 17:30:55 +02:00
parent 906c73b219
commit 725350d4ea
1 changed files with 2 additions and 0 deletions
@@ -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);