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);
}
}
}
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);