File size: 432 Bytes
0bd78bc bba3165 0bd78bc bba3165 0bd78bc bba3165 0bd78bc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
import torch
from torch import autocast
from diffusers import StableDiffusionPipeline
model_id = "CompVis/stable-diffusion-v1-4"
device = "cuda"
pipe = StableDiffusionPipeline.from_pretrained(model_id, use_auth_token=True)
pipe = pipe.to(device)
prompt = "a photo of an astronaut riding a horse on mars"
with autocast("cuda"):
image = pipe(prompt, guidance_scale=7.5).images[0]
image.save("astronaut_rides_horse.png") |