patrickvonplaten
commited on
Commit
•
c33f6f8
1
Parent(s):
91bc139
upload 0 images
Browse files- generated_image_pipeline.png +0 -0
- generated_image_unrolled.png +0 -0
- run.py +2 -2
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(
|
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(
|
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
|