Add Checkpoint Merger Pipeline

[InsaneOnion]

#271 Checkpoint Merger
生成结果对齐:
CompVis/stable-diffusion-v1-4 + runwayml/stable-diffusion-v1-5

torch

代码如下：

from diffusers import DiffusionPipeline
import torch

pipe = DiffusionPipeline.from_pretrained(
    "CompVis/stable-diffusion-v1-4",
    custom_pipeline="checkpoint_merger",
)

merged_pipe = pipe.merge(
    ["CompVis/stable-diffusion-v1-4", "runwayml/stable-diffusion-v1-5"],
    interp="sigmoid",
    alpha=0.4,
)

prompt = "An astronaut riding a horse on Mars"
merged_pipe.to("cuda")

image = merged_pipe(prompt, generator=torch.Generator("cuda").manual_seed(102)).images[0]
image.save("CompVis_runwayml.jpg")

paddle

代码如下：

from ppdiffusers import DiffusionPipeline
import paddle

pipe = DiffusionPipeline.from_pretrained(
    "CompVis/stable-diffusion-v1-4",
    custom_pipeline="/home/onion/workspace/code/pp/PaddleMIX/ppdiffusers/examples/community/checkpoint_merger",
)

merged_pipe = pipe.merge(
    ["CompVis/stable-diffusion-v1-4", "runwayml/stable-diffusion-v1-5"],
    interp="sigmoid",
    alpha=0.4,
)

prompt = "An astronaut riding a horse on Mars"

image = merged_pipe(prompt, generator=paddle.Generator("cuda").manual_seed(102)).images[0]
image.save("CompVis_runwayml.jpg")

结果一致

[paddle-bot]

Thanks for your contribution!

[westfish]

LGTM