텍스트에서 이미지 생성
확산 모델
설치
!pip install diffusers transformers accelerate scipy safetensors
임포트
from diffusers import DiffusionPipeline
모델 다운로드
generator = DiffusionPipeline.from_pretrained("CompVis/ldm-text2im-large-256")
GPU를 사용하도록 설정
generator.to("cuda")
생성
prompt = "An image of a squirrel in Picasso style"
result = generator([prompt], num_inference_steps=50, eta=0.3, guidance_scale=6)
0%| | 0/50 [00:00<?, ?it/s]
생성된 이미지
result.images[0]
