import torch
from diffusers import StableDiffusionPipeline
pipe = StableDiffusionPipeline.from_pretrained("sd-dreambooth-library/fashion")
pipe = pipe.to("cuda")
prompt = "a photograph of an astronaut riding a horse"
image = pipe(prompt).images[0]
image.save(f"astronaut_rides_horse.png")
image
image
de123af26a6c184d137487276175858.png
55aa13dfd7e61edbe61d1fba5affc0e.png 
