From bbd2fb719b3aa60ec875606ed47b8fec5bd69154 Mon Sep 17 00:00:00 2001
From: Aidova <ekaterina.aidova@intel.com>
Date: Sun, 29 Dec 2024 22:18:34 +0400
Subject: [PATCH] register dummy model class

---
 optimum/intel/__init__.py                             |  2 ++
 optimum/intel/openvino/__init__.py                    |  1 +
 .../utils/dummy_openvino_and_diffusers_objects.py     | 11 +++++++++++
 3 files changed, 14 insertions(+)

diff --git a/optimum/intel/__init__.py b/optimum/intel/__init__.py
index ad9fdca078..09dd456036 100644
--- a/optimum/intel/__init__.py
+++ b/optimum/intel/__init__.py
@@ -126,6 +126,7 @@
         "OVFluxPipeline",
         "OVFluxImg2ImgPipeline",
         "OVFluxInpaintPipeline",
+        "OVFluxFillPipeline"
         "OVPipelineForImage2Image",
         "OVPipelineForText2Image",
         "OVPipelineForInpainting",
@@ -148,6 +149,7 @@
             "OVFluxPipeline",
             "OVFluxImg2ImgPipeline",
             "OVFluxInpaintPipeline",
+            "OVFluxFillPieline",
             "OVPipelineForImage2Image",
             "OVPipelineForText2Image",
             "OVPipelineForInpainting",
diff --git a/optimum/intel/openvino/__init__.py b/optimum/intel/openvino/__init__.py
index 3737733329..e14b4ba003 100644
--- a/optimum/intel/openvino/__init__.py
+++ b/optimum/intel/openvino/__init__.py
@@ -84,6 +84,7 @@
         OVDiffusionPipeline,
         OVFluxImg2ImgPipeline,
         OVFluxInpaintPipeline,
+        OVFluxFillPipeline,
         OVFluxPipeline,
         OVLatentConsistencyModelImg2ImgPipeline,
         OVLatentConsistencyModelPipeline,
diff --git a/optimum/intel/utils/dummy_openvino_and_diffusers_objects.py b/optimum/intel/utils/dummy_openvino_and_diffusers_objects.py
index a6a10651cf..73c6d4d703 100644
--- a/optimum/intel/utils/dummy_openvino_and_diffusers_objects.py
+++ b/optimum/intel/utils/dummy_openvino_and_diffusers_objects.py
@@ -211,3 +211,14 @@ def __init__(self, *args, **kwargs):
     @classmethod
     def from_pretrained(cls, *args, **kwargs):
         requires_backends(cls, ["openvino", "diffusers"])
+
+
+class OVFluxFillPipeline(metaclass=DummyObject):
+    _backends = ["openvino", "diffusers"]
+
+    def __init__(self, *args, **kwargs):
+        requires_backends(self, ["openvino", "diffusers"])
+
+    @classmethod
+    def from_pretrained(cls, *args, **kwargs):
+        requires_backends(cls, ["openvino", "diffusers"])
\ No newline at end of file