Diffusion 활용해 간단한 이미지 생성

# 먼저 필요한 라이브러리가 설치되어 있어야 합니다:
# pip install torch diffusers transformers

import torch
from diffusers import StableDiffusionPipeline

# M1 CPU에서 실행하도록 설정 (MPS 사용 가능 시 활용)
device = "mps" if torch.backends.mps.is_available() else "cpu"

# Stable Diffusion 모델 로드
model_id = "runwayml/stable-diffusion-v1-5"
pipe = StableDiffusionPipeline.from_pretrained(model_id)
pipe = pipe.to(device)

# 프롬프트 설정
prompt = "A cute cat swimming in space"

# 이미지 생성
image = pipe(prompt).images[0]

# 이미지 저장
image.save("test_image.png")

<결과>

macos m1 32GB 메모리에서 최소 사양으로 Diffusion 모델을 이용하여 간단하게 이미지 생성하는 방법을 살펴 보았습니다.