Mirrors
/
diffusers
mirror of https://github.com/ShivamShrirao/diffusers.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

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

This commit is contained in:
Patrick von Platen 2022-10-19 15:54:07 +02:00 committed by GitHub
parent 6ea83608ad
commit 83b696e6c0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats 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, LMSDiscreteScheduler,
PNDMScheduler, PNDMScheduler,
StableDiffusionImg2ImgPipeline, StableDiffusionImg2ImgPipeline,
StableDiffusionInpaintPipeline, StableDiffusionInpaintPipelineLegacy,
StableDiffusionPipeline, StableDiffusionPipeline,
UNet2DConditionModel, UNet2DConditionModel,
) )
@ -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,