SDXL1.0の生ビール
# 1、パッケージのインストール
!pip install diffusers
!pip install invisible_watermark transformers accelerate safetensors
from diffusers import DiffusionPipeline
import torch
# 2、パイプラインの準備
pipe = DiffusionPipeline.from_pretrained(
"stabilityai/stable-diffusion-xl-base-1.0",
torch_dtype=torch.float16,
use_safetensors=True,
variant="fp16"
)
pipe.to("cuda")
# 3、画像生成の実行
prompt = "draft beer"
image = pipe(prompt=prompt).images[0]
# 保存と表示
image.save("output.png")
image
1→2→3の順で実行。GPUはT4。
呪文
呪文を見るにはログイン・会員登録が必須です。