mirror/whisper.cpp
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

whisper : set outputs from conv graph (#1959)

Browse Source
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
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
whisper.cpp
View File

@ -1680,6 +1680,8 @@ static struct ggml_cgraph * whisper_build_graph_conv(
wstate.embd_enc = cur; wstate.embd_enc = cur;
} }
ggml_set_output(cur);
ggml_build_forward_expand(gf, cur); ggml_build_forward_expand(gf, cur);
ggml_free(ctx0); ggml_free(ctx0);