from diffusers import StableDiffusionPipeline
import torch

# 加載基礎模型
pipe = StableDiffusionPipeline.from_pretrained(
    "stabilityai/stable-diffusion-xl-base-1.0", 
    torch_dtype=torch.float16
).to("cuda")

# 測試生成圖像
prompt = "A cute cartoon character in Chiikawa style, ultra-detailed"
image = pipe(prompt).images[0]
image.save("output.png")