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 ( ...@@ -10,7 +10,7 @@ from diffusers import (
LMSDiscreteScheduler, LMSDiscreteScheduler,
PNDMScheduler, PNDMScheduler,
StableDiffusionImg2ImgPipeline, StableDiffusionImg2ImgPipeline,
StableDiffusionInpaintPipeline, StableDiffusionInpaintPipelineLegacy,
StableDiffusionPipeline, StableDiffusionPipeline,
UNet2DConditionModel, UNet2DConditionModel,
) )
...@@ -136,7 +136,7 @@ class StableDiffusionMegaPipeline(DiffusionPipeline): ...@@ -136,7 +136,7 @@ class StableDiffusionMegaPipeline(DiffusionPipeline):
callback_steps: Optional[int] = 1, 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 # 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, prompt=prompt,
init_image=init_image, init_image=init_image,
mask_image=mask_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