File size: 285 Bytes
e18985e 20f0a97 e18985e 20f0a97 984edb7 20f0a97 9fca8be |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
from diffusers import StableDiffusionPipeline
DEVICE = 'mps'
pipe = StableDiffusionPipeline.from_pretrained(
"CompVis/stable-diffusion-v1-4",
use_auth_token=True
).to(DEVICE)
prompt = "Synthwave moonlight"
image = pipe(prompt).images[0]
image.save("pictures/Synthwave.png")
|