added an image save in text2image demo
#3
by
macadeliccc
- opened
README.md
CHANGED
@@ -59,6 +59,8 @@ prompt = "A photograph of the inside of a subway train. There are raccoons sitti
|
|
59 |
|
60 |
generator = torch.Generator(device="cpu").manual_seed(0)
|
61 |
image = pipe(prompt, num_inference_steps=25, generator=generator).images[0]
|
|
|
|
|
62 |
```
|
63 |
|
64 |
### Image-2-Image
|
|
|
59 |
|
60 |
generator = torch.Generator(device="cpu").manual_seed(0)
|
61 |
image = pipe(prompt, num_inference_steps=25, generator=generator).images[0]
|
62 |
+
|
63 |
+
image.save("kandinsky.png")
|
64 |
```
|
65 |
|
66 |
### Image-2-Image
|