From ae01322dbdf634bfec0265df347bf0196ba4eac4 Mon Sep 17 00:00:00 2001 From: "Yu, Zijun" Date: Mon, 8 Dec 2025 16:18:22 +0800 Subject: [PATCH] NPU fix wrong model output shape --- ggml/src/ggml-openvino/utils.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ggml/src/ggml-openvino/utils.cpp b/ggml/src/ggml-openvino/utils.cpp index ad99447f30..7412dcc2a8 100644 --- a/ggml/src/ggml-openvino/utils.cpp +++ b/ggml/src/ggml-openvino/utils.cpp @@ -614,8 +614,8 @@ ov::Tensor get_ov_output_tensor(std::shared_ptr ggml_decoder, con auto output_type = ggml_decoder->get_ov_type(ggml_tensor); auto output_shape = ggml_decoder->get_shape(ggml_tensor); - if (ggml_decoder->is_static() && result_name == "result_output") { - output_shape[1] = 1; + if (ggml_decoder->is_static() && result_name == "result_output" && output_shape[2] == 0) { + output_shape[2] = 1; } ov::Tensor output_tensor(output_type, output_shape, ggml_tensor->data); return output_tensor;