From a3c94f31e808c369f6a04399430faddb4c8cae5e Mon Sep 17 00:00:00 2001 From: Georgi Gerganov Date: Wed, 20 Mar 2024 18:22:40 +0200 Subject: [PATCH] whisper : minor --- whisper.cpp | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/whisper.cpp b/whisper.cpp index de3cfea85d1..67e3e06b555 100644 --- a/whisper.cpp +++ b/whisper.cpp @@ -2516,10 +2516,11 @@ static struct ggml_cgraph * whisper_build_graph_decoder( aheads_KQs = ggml_transpose(ctx0, aheads_KQs); aheads_KQs = ggml_cont(ctx0, aheads_KQs); aheads_KQs = ggml_reshape_3d(ctx0, aheads_KQs, KQ_soft_max->ne[0], KQ_soft_max->ne[1], wstate.aheads_masks.m[il]->ne[1]); - if (aheads_cross_QKs == NULL) + if (aheads_cross_QKs == NULL) { aheads_cross_QKs = aheads_KQs; - else + } else { aheads_cross_QKs = ggml_concat(ctx0, aheads_cross_QKs, aheads_KQs); + } } }