ddpm_dummy / README.md
patrickvonplaten's picture
Create README.md
9d0c1f2
|
raw
history blame
404 Bytes

Dummy diffusion model following architecture of https://github.com/lucidrains/denoising-diffusion-pytorch

Run the model as follows:

from diffusers import UNetModel
import torch

model = UNetModel.from_pretrained("fusing/ddpm_dummy")

batch_size, num_channels, height, width = 1, 3, 32, 32
dummy_noise = torch.tensor((batch_size, num_channels, height, width))

image = model(dummy_noise)