patrickvonplaten
commited on
Commit
•
c9de2e1
1
Parent(s):
a09a880
Update README.md
Browse files
README.md
CHANGED
@@ -9,7 +9,7 @@ import torch
|
|
9 |
model = UNetModel.from_pretrained("fusing/ddpm_dummy")
|
10 |
|
11 |
batch_size, num_channels, height, width = 1, 3, 32, 32
|
12 |
-
dummy_noise = torch.
|
13 |
time_step = torch.tensor([10])
|
14 |
|
15 |
image = model(dummy_noise, time_step)
|
|
|
9 |
model = UNetModel.from_pretrained("fusing/ddpm_dummy")
|
10 |
|
11 |
batch_size, num_channels, height, width = 1, 3, 32, 32
|
12 |
+
dummy_noise = torch.ones((batch_size, num_channels, height, width))
|
13 |
time_step = torch.tensor([10])
|
14 |
|
15 |
image = model(dummy_noise, time_step)
|