4컷 이미지 생성 모델

image.py

@@ -0,0 +1,19 @@
+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")