Skip to content

Commit dfc350a

Browse files
patrickvonplatenPrathik Rao
authored andcommitted
[Communit Pipeline] Make sure "mega" uses correct inpaint pipeline (huggingface#908)
1 parent a008713 commit dfc350a

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

examples/community/stable_diffusion_mega.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
LMSDiscreteScheduler,
1111
PNDMScheduler,
1212
StableDiffusionImg2ImgPipeline,
13-
StableDiffusionInpaintPipeline,
13+
StableDiffusionInpaintPipelineLegacy,
1414
StableDiffusionPipeline,
1515
UNet2DConditionModel,
1616
)
@@ -136,7 +136,7 @@ def inpaint(
136136
callback_steps: Optional[int] = 1,
137137
):
138138
# For more information on how this function works, please see: https://huggingface.co/docs/diffusers/api/pipelines/stable_diffusion#diffusers.StableDiffusionImg2ImgPipeline
139-
return StableDiffusionInpaintPipeline(**self.components)(
139+
return StableDiffusionInpaintPipelineLegacy(**self.components)(
140140
prompt=prompt,
141141
init_image=init_image,
142142
mask_image=mask_image,

0 commit comments

Comments
 (0)