diffusers scheduler add_noise前向加噪可视化_ddpm add_noise

参考：
http://www.bryh.cn/a/604194.html

1、diffusers Pipeline使用

import torch
from diffusers import PixArtAlphaPipeline

pipe = PixArtAlphaPipeline.from_pretrained("PixArt-alpha/PixArt-XL-2-1024-MS", torch_dtype=torch.float16)
pipe = pipe.to('cuda')

prompt = "A alpaca made of colorful building blocks, cyberpunk"
image = pipe(prompt, num_inference_steps=30).images[0]

image

image.save("./catcus.png") ##保存图片
1
2
3
4
5
6
7
8
9
10
11
12

加载lcm的gpu使用：

pipe = PixArtAlphaPipeline.from_pretrained("PixArt-alpha/PixArt-LCM-XL-2-1024-MS", torch_dtype=torch.float16)
pipe = pipe.to('cuda')
1
2