img2img pipeline

huggingface · Oct 7, 2024 · 689a757 · 689a757
1 parent f7f02c8
commit 689a757
Show file tree

Hide file tree

Showing 4 changed files with 352 additions and 5 deletions.
diff --git a/optimum/intel/__init__.py b/optimum/intel/__init__.py
@@ -253,6 +253,8 @@
     except OptionalDependencyNotAvailable:
         from .utils.dummy_openvino_and_diffusers_objects import (
             OVLatentConsistencyModelPipeline,
+            OVStableDiffusion3Img2ImgPipeline,
+            OVStableDiffusion3Pipeline,
             OVStableDiffusionImg2ImgPipeline,
             OVStableDiffusionInpaintPipeline,
             OVStableDiffusionPipeline,
@@ -262,6 +264,8 @@
     else:
         from .openvino import (
             OVLatentConsistencyModelPipeline,
+            OVStableDiffusion3Img2ImgPipeline,
+            OVStableDiffusion3Pipeline,
             OVStableDiffusionImg2ImgPipeline,
             OVStableDiffusionInpaintPipeline,
             OVStableDiffusionPipeline,

diff --git a/optimum/intel/openvino/__init__.py b/optimum/intel/openvino/__init__.py
@@ -82,6 +82,7 @@
 if is_diffusers_available():
     from .modeling_diffusion import (
         OVLatentConsistencyModelPipeline,
+        OVStableDiffusion3Img2ImgPipeline,
         OVStableDiffusion3Pipeline,
         OVStableDiffusionImg2ImgPipeline,
         OVStableDiffusionInpaintPipeline,