examples : adapt to new ggml_concat (#0)

ggerganov · May 29, 2024 · f911bed · f911bed
1 parent 2ae2db6
commit f911bed
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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);