Diffusers から芸術性の高い画像が作成できると噂の Playground v2.5 を使ってみる

結果

芸術性高いように感じます（適当）。

Pythonスクリプト

たったこれだけ。

from diffusers import DiffusionPipeline
import torch

pipe = DiffusionPipeline.from_pretrained(
    "playgroundai/playground-v2.5-1024px-aesthetic",
    torch_dtype=torch.float16,
    variant="fp16",
).to("cuda")

prompt = "a cute and beautiful girl, fountain background"
generator=torch.manual_seed(2024)
image = pipe(
    prompt=prompt,
    num_inference_steps=50,
    guidance_scale=3.0,
    generator=generator
).images[0]

image.save("result.png")