patrickvonplaten commited on
Commit
c33f6f8
1 Parent(s): 91bc139

upload 0 images

Browse files
generated_image_pipeline.png CHANGED
generated_image_unrolled.png CHANGED
run.py CHANGED
@@ -19,7 +19,7 @@ unet.to(torch_device)
19
  vqvae.to(torch_device)
20
 
21
  # generate gaussian noise to be decoded
22
- generator = torch.manual_seed(1)
23
  noise = torch.randn(
24
  (1, unet.in_channels, unet.image_size, unet.image_size),
25
  generator=generator,
@@ -64,7 +64,7 @@ import tqdm
64
  pipeline = LatentDiffusionUncondPipeline.from_pretrained("./")
65
 
66
  # generatae image by calling the pipeline
67
- generator = torch.manual_seed(1)
68
  image = pipeline(generator=generator, num_inference_steps=200)["sample"]
69
 
70
  # process image
 
19
  vqvae.to(torch_device)
20
 
21
  # generate gaussian noise to be decoded
22
+ generator = torch.manual_seed(0)
23
  noise = torch.randn(
24
  (1, unet.in_channels, unet.image_size, unet.image_size),
25
  generator=generator,
 
64
  pipeline = LatentDiffusionUncondPipeline.from_pretrained("./")
65
 
66
  # generatae image by calling the pipeline
67
+ generator = torch.manual_seed(0)
68
  image = pipeline(generator=generator, num_inference_steps=200)["sample"]
69
 
70
  # process image