[Python] Added VLMPipeline::set_chat_template (#1097)

openvinotoolkit · Oct 29, 2024 · 86782e1 · 86782e1
1 parent 9daaf5a
commit 86782e1
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/python/py_vlm_pipeline.cpp b/src/python/py_vlm_pipeline.cpp
@@ -133,6 +133,7 @@ void init_vlm_pipeline(py::module_& m) {
 
         .def("start_chat", &ov::genai::VLMPipeline::start_chat, py::arg("system_message") = "")
         .def("finish_chat", &ov::genai::VLMPipeline::finish_chat)
+        .def("set_chat_template", &ov::genai::VLMPipeline::set_chat_template)
         .def("get_tokenizer", &ov::genai::VLMPipeline::get_tokenizer)
         .def("get_generation_config", &ov::genai::VLMPipeline::get_generation_config)
         .def("set_generation_config", &ov::genai::VLMPipeline::set_generation_config)