AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

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

Browse Source
This commit is contained in:
Patrick von Platen
2022-10-19 15:54:07 +02:00
committed by GitHub
parent 6ea83608ad
commit 83b696e6c0
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/community/stable_diffusion_mega.py
View File

@@ -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,