macadeliccc
commited on
Commit
·
2663b39
1
Parent(s):
0e855b0
added an image save in text2image demo
Browse filesuser cant see the generated image because it doesnt display anywhere and is destroyed after the python context is over
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
|