Update README.md
Browse files
README.md
CHANGED
|
@@ -20,6 +20,7 @@ pipe.to(device)
|
|
| 20 |
# run pipeline in inference (sample random noise and denoise)
|
| 21 |
generator = torch.Generator(device=device).manual_seed(46)
|
| 22 |
image = pipe(
|
|
|
|
| 23 |
batch_size=1,
|
| 24 |
num_inference_steps=1000
|
| 25 |
).images
|
|
|
|
| 20 |
# run pipeline in inference (sample random noise and denoise)
|
| 21 |
generator = torch.Generator(device=device).manual_seed(46)
|
| 22 |
image = pipe(
|
| 23 |
+
generator=generator,
|
| 24 |
batch_size=1,
|
| 25 |
num_inference_steps=1000
|
| 26 |
).images
|