Update README.md
Browse files
README.md
CHANGED
@@ -36,7 +36,7 @@ pipe.unet = unet
|
|
36 |
pipe = pipe.to("cuda")
|
37 |
|
38 |
prompt = "Two cats playing chess on a tree branch"
|
39 |
-
image = pipe(prompt, guidance_scale=
|
40 |
|
41 |
image.save("cats_playing_chess.png")
|
42 |
```
|
|
|
36 |
pipe = pipe.to("cuda")
|
37 |
|
38 |
prompt = "Two cats playing chess on a tree branch"
|
39 |
+
image = pipe(prompt, guidance_scale=5).images[0].resize((512,512))
|
40 |
|
41 |
image.save("cats_playing_chess.png")
|
42 |
```
|