diffusers
/

ddpm_dummy

Model card Files Files and versions

patrickvonplaten commited on May 31, 2022

Commit

c9de2e1

·

1 Parent(s): a09a880

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -9,7 +9,7 @@ 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))
 time_step = torch.tensor([10])
 image = model(dummy_noise, time_step)

 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)