from diffusers import StableDiffusionPipeline
import torch

def generate_sd_turbo(prompt: str):
    model_id = "stabilityai/sd-turbo"

    pipe = StableDiffusionPipeline.from_pretrained(
        model_id,
        torch_dtype=torch.float16,
        use_safetensors=True
    ).to("cuda")

    image = pipe(prompt=prompt, guidance_scale=0.0).images[0]
    image.save("output_sd_turbo.png")
    return image

if __name__ == "__main__":
    prompt = "그녀를 바라보는 한 남자의 야망"
    img = generate_sd_turbo(prompt)
    print("✅ 저장 완료: output_sd_turbo.png")
    img.show()