Skip to content

Commit 5a9b886

Browse files
committed
register dummy model class
1 parent 4be7cd2 commit 5a9b886

File tree

3 files changed

+14
-0
lines changed

3 files changed

+14
-0
lines changed

optimum/intel/__init__.py

+2
Original file line numberDiff line numberDiff line change
@@ -126,6 +126,7 @@
126126
"OVFluxPipeline",
127127
"OVFluxImg2ImgPipeline",
128128
"OVFluxInpaintPipeline",
129+
"OVFluxFillPipeline",
129130
"OVPipelineForImage2Image",
130131
"OVPipelineForText2Image",
131132
"OVPipelineForInpainting",
@@ -148,6 +149,7 @@
148149
"OVFluxPipeline",
149150
"OVFluxImg2ImgPipeline",
150151
"OVFluxInpaintPipeline",
152+
"OVFluxFillPipeline",
151153
"OVPipelineForImage2Image",
152154
"OVPipelineForText2Image",
153155
"OVPipelineForInpainting",

optimum/intel/openvino/__init__.py

+1
Original file line numberDiff line numberDiff line change
@@ -82,6 +82,7 @@
8282
if is_diffusers_available():
8383
from .modeling_diffusion import (
8484
OVDiffusionPipeline,
85+
OVFluxFillPipeline,
8586
OVFluxImg2ImgPipeline,
8687
OVFluxInpaintPipeline,
8788
OVFluxPipeline,

optimum/intel/utils/dummy_openvino_and_diffusers_objects.py

+11
Original file line numberDiff line numberDiff line change
@@ -211,3 +211,14 @@ def __init__(self, *args, **kwargs):
211211
@classmethod
212212
def from_pretrained(cls, *args, **kwargs):
213213
requires_backends(cls, ["openvino", "diffusers"])
214+
215+
216+
class OVFluxFillPipeline(metaclass=DummyObject):
217+
_backends = ["openvino", "diffusers"]
218+
219+
def __init__(self, *args, **kwargs):
220+
requires_backends(self, ["openvino", "diffusers"])
221+
222+
@classmethod
223+
def from_pretrained(cls, *args, **kwargs):
224+
requires_backends(cls, ["openvino", "diffusers"])

0 commit comments

Comments
 (0)