Unverified Commit 83b696e6 authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

[Communit Pipeline] Make sure "mega" uses correct inpaint pipeline (#908)

parent 6ea83608
......@@ -10,7 +10,7 @@ from diffusers import (
LMSDiscreteScheduler,
PNDMScheduler,
StableDiffusionImg2ImgPipeline,
StableDiffusionInpaintPipeline,
StableDiffusionInpaintPipelineLegacy,
StableDiffusionPipeline,
UNet2DConditionModel,
)
......@@ -136,7 +136,7 @@ class StableDiffusionMegaPipeline(DiffusionPipeline):
callback_steps: Optional[int] = 1,
):
# For more information on how this function works, please see: https://huggingface.co/docs/diffusers/api/pipelines/stable_diffusion#diffusers.StableDiffusionImg2ImgPipeline
return StableDiffusionInpaintPipeline(**self.components)(
return StableDiffusionInpaintPipelineLegacy(**self.components)(
prompt=prompt,
init_image=init_image,
mask_image=mask_image,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment