image.py

from diffusers import StableDiffusionXLPipeline
import torch

# 모델 로드 (SDXL Base 1.0)
pipe = StableDiffusionXLPipeline.from_pretrained(
    "stabilityai/stable-diffusion-xl-base-1.0", 
    torch_dtype=torch.float16,
    variant="fp16",
    use_safetensors=True
).to("cuda")

# 텍스트 프롬프트
prompt = '그녀를 바라보는 한 남자의 야망'


# 이미지 생성
image = pipe(prompt=prompt).images[0]
image.save("output_scene2.png")
print("✅ 이미지 저장 완료: output_scene2.png")
print("확인용")