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