whisper : set outputs from conv graph (#1959)

This commit is contained in:
Georgi Gerganov 2024-03-16 17:30:55 +02:00
parent 906c73b219
commit 725350d4ea
No known key found for this signature in database
GPG Key ID: 449E073F9DC10735

View File

@ -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);