examples : adapt to new ggml_concat (ggml/0)
This commit is contained in:
parent
109148ac84
commit
02fc147a0b
@ -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);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user