diff --git a/optimum/intel/__init__.py b/optimum/intel/__init__.py index ad9fdca07..09dd45603 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 373773332..e14b4ba00 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 a6a10651c..73c6d4d70 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