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) {
aheads_cross_QKs = aheads_KQs;
} else {
aheads_cross_QKs = ggml_concat(ctx0, aheads_cross_QKs, aheads_KQs);
aheads_cross_QKs = ggml_concat(ctx0, aheads_cross_QKs, aheads_KQs, 2);
}
}
}