From: Georgi Gerganov Date: Wed, 29 May 2024 09:58:00 +0000 (+0300) Subject: examples : adapt to new ggml_concat (#0) X-Git-Tag: upstream/0.0.1642~633 X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=f911bede8783988ee6ab1a38093b22e854692f99;p=pkg%2Fggml%2Fsources%2Fggml examples : adapt to new ggml_concat (#0) --- diff --git a/examples/whisper/whisper.cpp b/examples/whisper/whisper.cpp index 84aec823..db828f54 100644 --- a/examples/whisper/whisper.cpp +++ b/examples/whisper/whisper.cpp @@ -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); } } } diff --git a/examples/yolo/yolov3-tiny.cpp b/examples/yolo/yolov3-tiny.cpp index 2467bd89..b8c81347 100644 --- a/examples/yolo/yolov3-tiny.cpp +++ b/examples/yolo/yolov3-tiny.cpp @@ -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);