superdiff
/

superdiff-sdxl-v1-0

SuperDiffSDXLPipeline

Model card Files Files and versions

mskrt commited on Feb 18

Commit

b5d138c

·

verified ·

1 Parent(s): 2974fbc

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -41,7 +41,7 @@ from diffusers import DiffusionPipeline
 pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", custom_pipeline="superdiff/superdiff-sdxl-v1-0")
 output = pipeline("a flamingo", "a candy cane", seed=1, num_inference_steps=200, batch_size=1)
-image = Image.fromarray(output[0].cpu().numpy())
 image.save("superdiff_output.png")
 ```

 pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", custom_pipeline="superdiff/superdiff-sdxl-v1-0")
 output = pipeline("a flamingo", "a candy cane", seed=1, num_inference_steps=200, batch_size=1)
+image = Image.fromarray(output[0])
 image.save("superdiff_output.png")
 ```