diffusers
/

ddpm_dummy

Model card Files Files and versions

patrickvonplaten commited on May 31, 2022

Commit

a09a880

·

1 Parent(s): 9d0c1f2

Update README.md

Files changed (1) hide show

README.md +2 -1

README.md CHANGED Viewed

@@ -10,6 +10,7 @@ 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)
 ```

 batch_size, num_channels, height, width = 1, 3, 32, 32
 dummy_noise = torch.tensor((batch_size, num_channels, height, width))
+time_step = torch.tensor([10])
+image = model(dummy_noise, time_step)
 ```