Updating example code in README.md
#118
by
Jinkura
- opened
README.md
CHANGED
@@ -57,7 +57,7 @@ pipe = StableDiffusionPipeline.from_pretrained(
|
|
57 |
|
58 |
prompt = "1girl, aqua eyes, baseball cap, blonde hair, closed mouth, earrings, green background, hat, hoop earrings, jewelry, looking at viewer, shirt, short hair, simple background, solo, upper body, yellow shirt"
|
59 |
with autocast("cuda"):
|
60 |
-
image = pipe(prompt, guidance_scale=6)[
|
61 |
|
62 |
image.save("test.png")
|
63 |
```
|
|
|
57 |
|
58 |
prompt = "1girl, aqua eyes, baseball cap, blonde hair, closed mouth, earrings, green background, hat, hoop earrings, jewelry, looking at viewer, shirt, short hair, simple background, solo, upper body, yellow shirt"
|
59 |
with autocast("cuda"):
|
60 |
+
image = pipe(prompt, guidance_scale=6).images[0]
|
61 |
|
62 |
image.save("test.png")
|
63 |
```
|