examples : adapt to new ggml_concat (ggml/0)

This commit is contained in:
Georgi Gerganov
2024-06-16 18:19:48 +03:00
parent 109148ac84
commit 02fc147a0b
+1 -1
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);
}
}
}