From a3be810bd8110fecddeca61d40ce436b9cb6f50a Mon Sep 17 00:00:00 2001 From: Ilya Lavrenov Date: Fri, 10 Jan 2025 12:16:20 +0400 Subject: [PATCH] Python: reflect inheritance between VLM and LLM decoded results (#1519) --- src/python/openvino_genai/py_openvino_genai.pyi | 2 +- src/python/py_vlm_pipeline.cpp | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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)