File size: 444 Bytes
9d0c1f2 c9de2e1 a09a880 9d0c1f2 a09a880 9d0c1f2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
# Dummy diffusion model following architecture of https://github.com/lucidrains/denoising-diffusion-pytorch
Run the model as follows:
```python
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.ones((batch_size, num_channels, height, width))
time_step = torch.tensor([10])
image = model(dummy_noise, time_step)
``` |