Skip to content

Commit f911bed

Browse files
committed
examples : adapt to new ggml_concat (#0)
1 parent 2ae2db6 commit f911bed

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

examples/whisper/whisper.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2588,7 +2588,7 @@ static struct ggml_cgraph * whisper_build_graph_decoder(
25882588
if (aheads_cross_QKs == NULL) {
25892589
aheads_cross_QKs = aheads_KQs;
25902590
} else {
2591-
aheads_cross_QKs = ggml_concat(ctx0, aheads_cross_QKs, aheads_KQs);
2591+
aheads_cross_QKs = ggml_concat(ctx0, aheads_cross_QKs, aheads_KQs, 2);
25922592
}
25932593
}
25942594
}

examples/yolo/yolov3-tiny.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -412,7 +412,7 @@ void detect(yolo_image & img, const yolo_model & model, float thresh, const std:
412412
print_shape(18, result);
413413
result = ggml_upscale(ctx0, result, 2);
414414
print_shape(19, result);
415-
result = ggml_concat(ctx0, result, layer_8);
415+
result = ggml_concat(ctx0, result, layer_8, 2);
416416
print_shape(20, result);
417417
result = apply_conv2d(ctx0, result, model.conv2d_layers[11]);
418418
print_shape(21, result);

0 commit comments

Comments
 (0)