hakurei
/

waifu-diffusion

StableDiffusionPipeline

stable-diffusion

Inference Endpoints

Model card Files Files and versions Community

Fix README.md "Example Code"

#99

by Krk97 - opened Dec 31, 2022

base: refs/heads/main

←

from: refs/pr/99

Discussion Files changed

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -57,7 +57,7 @@ pipe = StableDiffusionPipeline.from_pretrained(
 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"
 with autocast("cuda"):
-    image = pipe(prompt, guidance_scale=6)["sample"][0]
 image.save("test.png")
 ```

 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"
 with autocast("cuda"):
+    image = pipe(prompt, guidance_scale=6)["images"][0]
 image.save("test.png")
 ```