save

Author: dkalinowski

src/image_gen/BUILD

@@ -38,6 +38,21 @@ cc_library(
     alwayslink = 1,
 )
 
+cc_library(
+    name = "pipelines",
+    hdrs = ["pipelines.hpp"],
+    srcs = ["pipelines.cpp"],
+    deps = select({
+        "//conditions:default": ["//third_party:genai", ":llm_engine"],
+        "//:not_genai_bin" : [":llm_engine"],
+    }),
+    visibility = ["//visibility:public"],
+    local_defines = COMMON_LOCAL_DEFINES,
+    copts = COPTS_ADJUSTED,
+    linkopts = LINKOPTS_ADJUSTED,
+    alwayslink = 1,
+)
+
 cc_library(
     name = "image_gen_calculator",
     srcs = ["http_image_gen_calculator.cc"],

src/image_gen/http_image_gen_calculator.cc

@@ -29,6 +29,8 @@ using namespace ovms;
 
 namespace mediapipe {
 
+const std::string IMAGE_GEN_SESSION_SIDE_PACKET_TAG = "IMAGE_GEN_NODE_RESOURCES";
+
 class HttpImageGenCalculator : public CalculatorBase {
     static const std::string INPUT_TAG_NAME;
     static const std::string OUTPUT_TAG_NAME;

src/image_gen/pipelines.cpp

@@ -0,0 +1,25 @@
+//*****************************************************************************
+// Copyright 2025 Intel Corporation
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//*****************************************************************************
+#include "pipelines.hpp"
+
+namespace ovms {
+
+ImageGenerationPipelines::ImageGenerationPipelines(const std::string& models_path) :
+    text2ImagePipeline(ov::genai::Text2ImagePipeline(models_path/*TODO Device*/)) {
+    // TODO: Make other pipelines out of the basic one, with shared models, GenAI API supports that
+}
+
+}  // namespace ovms

src/image_gen/pipelines.hpp

@@ -0,0 +1,35 @@
+//*****************************************************************************
+// Copyright 2025 Intel Corporation
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//*****************************************************************************
+#pragma once
+
+#include <string>
+
+#include <openvino/genai/image_generation/text2image_pipeline.hpp>
+
+namespace ovms {
+
+struct ImageGenerationPipelines {
+    ov::genai::Text2ImagePipeline text2ImagePipeline;
+
+    ImageGenerationPipelines() = delete;
+    ImageGenerationPipelines(const std::string& models_path);
+    ImageGenerationPipelines(const ImageGenerationPipelines&) = delete;
+    ImageGenerationPipelines& operator=(const ImageGenerationPipelines&) = delete;
+    ImageGenerationPipelines(ImageGenerationPipelines&&) = delete;
+    ImageGenerationPipelines& operator=(ImageGenerationPipelines&&) = delete;
+};
+
+}  // namespace ovms

src/mediapipe_internal/mediapipegraphexecutor.cpp

@@ -60,6 +60,7 @@ MediapipeGraphExecutor::MediapipeGraphExecutor(
 
 const std::string MediapipeGraphExecutor::PYTHON_SESSION_SIDE_PACKET_TAG = "py";
 const std::string MediapipeGraphExecutor::LLM_SESSION_SIDE_PACKET_TAG = "llm";
+const std::string MediapipeGraphExecutor::IMAGE_GEN_SESSION_SIDE_PACKET_TAG = "pipes";
 const ::mediapipe::Timestamp MediapipeGraphExecutor::STARTING_TIMESTAMP = ::mediapipe::Timestamp(0);
 
 }  // namespace ovms

src/mediapipe_internal/mediapipegraphexecutor.hpp

@@ -91,6 +91,7 @@ class MediapipeGraphExecutor {
 public:
     static const std::string PYTHON_SESSION_SIDE_PACKET_TAG;
     static const std::string LLM_SESSION_SIDE_PACKET_TAG;
+    static const std::string IMAGE_GEN_SESSION_SIDE_PACKET_TAG;
     static const ::mediapipe::Timestamp STARTING_TIMESTAMP;
 
     MediapipeGraphExecutor(const std::string& name, const std::string& version, const ::mediapipe::CalculatorGraphConfig& config,