Update README.md
Browse files
README.md
CHANGED
@@ -41,7 +41,7 @@ image = pipe(
|
|
41 |
width=1024,
|
42 |
num_inference_steps=50,
|
43 |
guidance_scale=5.0,
|
44 |
-
generator=
|
45 |
).images[0]
|
46 |
image.show()
|
47 |
```
|
|
|
41 |
width=1024,
|
42 |
num_inference_steps=50,
|
43 |
guidance_scale=5.0,
|
44 |
+
generator=torch.Generator(pipe.device).manual_seed(66),
|
45 |
).images[0]
|
46 |
image.show()
|
47 |
```
|