Updating example code in README.md

According to this [issue](https://github.com/CompVis/stable-diffusion/issues/402) we should now use `.images` field instead of the `["sample"]` key.

Files changed (1) hide show

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