]> git.djapps.eu Git - pkg/ggml/sources/ggml/commitdiff
examples : adapt to new ggml_concat (#0)
authorGeorgi Gerganov <redacted>
Wed, 29 May 2024 09:58:00 +0000 (12:58 +0300)
committerGeorgi Gerganov <redacted>
Wed, 29 May 2024 10:16:38 +0000 (13:16 +0300)
examples/whisper/whisper.cpp
examples/yolo/yolov3-tiny.cpp

index 84aec8238cdb42d19cab3ef2e97a5aa0b91a2a94..db828f5417b14d64f784f64458fb92b62846cd75 100644 (file)
@@ -2588,7 +2588,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);
                         }
                     }
                 }
index 2467bd89f33fa3d254701d30bceb43d30001da0f..b8c81347c2e6f6e4a5307c042cbdabbf42e12ec4 100644 (file)
@@ -412,7 +412,7 @@ void detect(yolo_image & img, const yolo_model & model, float thresh, const std:
     print_shape(18, result);
     result = ggml_upscale(ctx0, result, 2);
     print_shape(19, result);
-    result = ggml_concat(ctx0, result, layer_8);
+    result = ggml_concat(ctx0, result, layer_8, 2);
     print_shape(20, result);
     result = apply_conv2d(ctx0, result, model.conv2d_layers[11]);
     print_shape(21, result);