diff --git a/src/python/openvino_genai/py_openvino_genai.pyi b/src/python/openvino_genai/py_openvino_genai.pyi index 5adde32db4..4145aa7620 100644 --- a/src/python/openvino_genai/py_openvino_genai.pyi +++ b/src/python/openvino_genai/py_openvino_genai.pyi @@ -1745,7 +1745,7 @@ class UNet2DConditionModel: ... def set_hidden_states(self, tensor_name: str, encoder_hidden_states: openvino._pyopenvino.Tensor) -> None: ... -class VLMDecodedResults: +class VLMDecodedResults(DecodedResults): """ Structure to store resulting batched text outputs and scores for each batch. diff --git a/src/python/py_vlm_pipeline.cpp b/src/python/py_vlm_pipeline.cpp index 570018f341..e076d87f11 100644 --- a/src/python/py_vlm_pipeline.cpp +++ b/src/python/py_vlm_pipeline.cpp @@ -112,7 +112,7 @@ void init_vlm_pipeline(py::module_& m) { .def("get_prepare_embeddings_duration", &ov::genai::VLMPerfMetrics::get_prepare_embeddings_duration) .def_readonly("vlm_raw_metrics", &ov::genai::VLMPerfMetrics::vlm_raw_metrics); - py::class_(m, "VLMDecodedResults", decoded_results_docstring) + py::class_(m, "VLMDecodedResults", decoded_results_docstring) .def(py::init<>()) .def_property_readonly("texts", [](const ov::genai::VLMDecodedResults &dr) -> py::typing::List { return pyutils::handle_utf8(dr.texts); }) .def_readonly("scores", &ov::genai::VLMDecodedResults::scores)