Update README.md
Browse files
README.md
CHANGED
|
@@ -38,9 +38,8 @@ pipe = DiffusionPipeline.from_pretrained(
|
|
| 38 |
#
|
| 39 |
batch_size = 1
|
| 40 |
in_channels = 6
|
| 41 |
-
sample_size = 160
|
| 42 |
sequence_len = 3
|
| 43 |
|
|
|
|
| 44 |
lres_image = torch.randn((sequence_len, in_channels, sample_size, sample_size)).to(pipe.device)
|
| 45 |
-
|
| 46 |
-
pipe(image =inputs)
|
|
|
|
| 38 |
#
|
| 39 |
batch_size = 1
|
| 40 |
in_channels = 6
|
|
|
|
| 41 |
sequence_len = 3
|
| 42 |
|
| 43 |
+
#
|
| 44 |
lres_image = torch.randn((sequence_len, in_channels, sample_size, sample_size)).to(pipe.device)
|
| 45 |
+
pipe(image=lres_image)
|
|
|