Hugging Faceのstable-diffusion-xl-base-1.0を使ってみました。
1 準備
Google Colabで”ファイル/ノートブックを新規作成”し、 "ランタイム/ランタイムのタイプを変更"で、"T4 GPU"を選択します。
2 インストール
diffusersをインストール。 加えて、invisible watermark、transformers、accelerate、safetensorsをインストール。!pip install diffusers --upgrade
!pip install invisible_watermark transformers accelerate safetensors
3 コードを記述
DiffusionPipelineに、"stabilityai/stable-diffusion-xl-base-1.0"を指定したコードを実行すると、ベースモデルを使った画像が生成されます。画像は、Google Driveに保存します。
from diffusers import DiffusionPipeline
from accelerate.utils import write_basic_config
import torch
from google.colab import drive
write_basic_config()
drive.mount("/content/drive")
pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16, use_safetensors=True, variant="fp16")
pipe.to("cuda")
prompt = "An astronaut riding a green horse"
#prompt = "zebra in Africa"
image = pipe(prompt=prompt).images[0]
image.save("/content/drive/MyDrive/outputimage.png")
4 実行
コードを実行すると、Google Driveにpromptで指定した画像が出力されます。