Dev support stable diffusion inpainting (huggingface#71)

Author: Flowingsun007

src/diffusers/__init__.py

@@ -86,6 +86,7 @@
 else:
     from .utils.dummy_torch_and_transformers_objects import *  # noqa F403
 
+
 if is_torch_available() and is_transformers_available() and is_onnx_available():
     from .pipelines import (
         OnnxStableDiffusionImg2ImgPipeline,
@@ -141,5 +142,6 @@
 from .pipelines import OneFlowAltDiffusionPipeline
 from .pipelines import OneFlowStableDiffusionPipeline
 from .pipelines import OneFlowStableDiffusionImg2ImgPipeline
+from .pipelines import OneFlowStableDiffusionInpaintPipeline, OneFlowStableDiffusionInpaintPipelineLegacy
 from .pipeline_oneflow_utils import OneFlowDiffusionPipeline
 from .modeling_oneflow_utils import OneFlowModelMixin

src/diffusers/pipeline_oneflow_utils.py

@@ -561,7 +561,7 @@ def from_pretrained(cls, pretrained_model_name_or_path: Optional[Union[str, os.P
         if pipeline_class.__name__ == "StableDiffusionInpaintPipeline" and version.parse(
             version.parse(config_dict["_diffusers_version"]).base_version
         ) <= version.parse("0.5.1"):
-            from diffusers import StableDiffusionInpaintPipeline, StableDiffusionInpaintPipelineLegacy
+            from diffusers import OneFlowStableDiffusionInpaintPipeline as StableDiffusionInpaintPipeline, OneFlowStableDiffusionInpaintPipelineLegacy as StableDiffusionInpaintPipelineLegacy
 
             pipeline_class = StableDiffusionInpaintPipelineLegacy
 

src/diffusers/pipelines/__init__.py

@@ -50,3 +50,4 @@
 from .alt_diffusion import OneFlowAltDiffusionPipeline
 from .stable_diffusion import OneFlowStableDiffusionPipeline
 from .stable_diffusion import OneFlowStableDiffusionImg2ImgPipeline
+from .stable_diffusion import OneFlowStableDiffusionInpaintPipeline, OneFlowStableDiffusionInpaintPipelineLegacy

src/diffusers/pipelines/stable_diffusion/__init__.py

@@ -79,4 +79,6 @@ class FlaxStableDiffusionPipelineOutput(BaseOutput):
 
 from .pipeline_stable_diffusion_oneflow import OneFlowStableDiffusionPipeline
 from .pipeline_stable_diffusion_img2img_oneflow import OneFlowStableDiffusionImg2ImgPipeline
+from .pipeline_stable_diffusion_inpaint_oneflow import OneFlowStableDiffusionInpaintPipeline
+from .pipeline_stable_diffusion_inpaint_legacy_oneflow import OneFlowStableDiffusionInpaintPipelineLegacy
 from .safety_checker_oneflow import OneFlowStableDiffusionSafetyChecker