파이션 Stable Diffusion 초 간단 예제

from diffusers import StableDiffusionPipeline

pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4")

pipe = pipe.to("cuda")

prompt = "a photo of an austonaut riding a horse on mars"

image = pipe(prompt).images[0]

image.save(astronaut_rides_horse.png")

 

 

<< 실행 조건 >>

• pip로 패키지 설치
• huggingface-cli 에 반드시 로그인해야 함
• 모델 다운로드 받아 경로 지정 필요