]> git.djapps.eu Git - pkg/ggml/sources/whisper.cpp/commitdiff
examples : adapt to new ggml_concat (ggml/0)
authorGeorgi Gerganov <redacted>
Wed, 29 May 2024 09:58:00 +0000 (12:58 +0300)
committerGeorgi Gerganov <redacted>
Sun, 16 Jun 2024 15:19:48 +0000 (18:19 +0300)
whisper.cpp

index 71eb06733ec06fb5d63065207152eaeca4b80be0..0a53a03d03d8dd5f28d0702b93dcbd833bdd6996 100644 (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);
                         }
                     }
                 }