hakurei
/

waifu-diffusion

StableDiffusionPipeline

stable-diffusion

Inference Endpoints

Model card Files Files and versions Community

Updating the diffuser image output

#120

by abhishekbhakat - opened Mar 25, 2023

base: refs/heads/main

←

from: refs/pr/120

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")
 ```