Skip to content

Commit e8aaedf

Browse files
authored
VisionEncoderPhi3V: fix multiple infers (openvinotoolkit#1914)
1 parent 2b94f73 commit e8aaedf

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/cpp/src/visual_language/phi3_vision/classes.cpp

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -218,8 +218,10 @@ EncodedImage VisionEncoderPhi3V::encode(const ov::Tensor& image, const ov::AnyMa
218218

219219
const auto& [pixel_values, image_size] = get_pixel_values_phi3_v(image, config);
220220
encoder.set_input_tensor(pixel_values);
221+
ov::Tensor res{ov::element::f32, encoder.get_output_tensor().get_shape()};
222+
encoder.set_output_tensor(res);
221223
encoder.infer();
222-
return {encoder.get_output_tensor(), image_size};
224+
return {std::move(res), image_size};
223225
}
224226

225227
namespace {

0 commit comments

Comments
 (0)