license: apache-2.0 | |
import torch | |
from diffusers import StableDiffusionPipeline | |
pipe = StableDiffusionPipeline.from_pretrained("sd-dreambooth-library/fashsion") | |
prompt = "a photograph of an astronaut riding a horse" | |
image = pipe(prompt).images[0] | |
image.save(f"astronaut_rides_horse.png") | |
![image](https://s3.amazonaws.com/moonup/production/uploads/1669885576067-63044d493926de1f7ec709f4.png) | |