Skip to content

Commit a3be810

Browse files
Python: reflect inheritance between VLM and LLM decoded results (openvinotoolkit#1519)
1 parent 12a52e3 commit a3be810

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

src/python/openvino_genai/py_openvino_genai.pyi

+1-1
Original file line numberDiff line numberDiff line change
@@ -1745,7 +1745,7 @@ class UNet2DConditionModel:
17451745
...
17461746
def set_hidden_states(self, tensor_name: str, encoder_hidden_states: openvino._pyopenvino.Tensor) -> None:
17471747
...
1748-
class VLMDecodedResults:
1748+
class VLMDecodedResults(DecodedResults):
17491749
"""
17501750
17511751
Structure to store resulting batched text outputs and scores for each batch.

src/python/py_vlm_pipeline.cpp

+1-1
Original file line numberDiff line numberDiff line change
@@ -112,7 +112,7 @@ void init_vlm_pipeline(py::module_& m) {
112112
.def("get_prepare_embeddings_duration", &ov::genai::VLMPerfMetrics::get_prepare_embeddings_duration)
113113
.def_readonly("vlm_raw_metrics", &ov::genai::VLMPerfMetrics::vlm_raw_metrics);
114114

115-
py::class_<ov::genai::VLMDecodedResults>(m, "VLMDecodedResults", decoded_results_docstring)
115+
py::class_<ov::genai::VLMDecodedResults, ov::genai::DecodedResults>(m, "VLMDecodedResults", decoded_results_docstring)
116116
.def(py::init<>())
117117
.def_property_readonly("texts", [](const ov::genai::VLMDecodedResults &dr) -> py::typing::List<py::str> { return pyutils::handle_utf8(dr.texts); })
118118
.def_readonly("scores", &ov::genai::VLMDecodedResults::scores)

0 commit comments

Comments
 (0)