lschmidt commited on
Commit
1a5843d
·
verified ·
1 Parent(s): aab5a27

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -3
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
- print(inputs.shape)
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)