examples : adapt to new ggml_concat (ggml/0)

This commit is contained in:
Georgi Gerganov 2024-05-29 12:58:00 +03:00
parent 109148ac84
commit 02fc147a0b

View File

@ -2582,7 +2582,7 @@ static struct ggml_cgraph * whisper_build_graph_decoder(
if (aheads_cross_QKs == NULL) { if (aheads_cross_QKs == NULL) {
aheads_cross_QKs = aheads_KQs; aheads_cross_QKs = aheads_KQs;
} else { } else {
aheads_cross_QKs = ggml_concat(ctx0, aheads_cross_QKs, aheads_KQs); aheads_cross_QKs = ggml_concat(ctx0, aheads_cross_QKs, aheads_KQs, 2);
} }
} }
} }