· Read in about 1 min · (309 words) ·

Stable DiffusionをGoogle Colab上で使ってみました。 コードとしては以下のような形です。bashのコードはJupyterから投げます。

pip install diffusers==0.2.4
pip install transformers scipy ftfy
pip install "ipywidgets>=7,<8"
from google.colab import output
from huggingface_hub import notebook_login

import torch
from diffusers import StableDiffusionPipeline

# make sure you're logged in with `huggingface-cli login`
pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", revision="fp16", torch_dtype=torch.float16, use_auth_token=True)
pipe ="cuda")
from torch import autocast

prompt = "a galaxy far from earth"
with autocast("cuda"):
  image = pipe(prompt)["sample"][0]  # image here is in [PIL format](

# Now to display an image you can do either save it such as:"galaxy_far_from_earth.png")

# or if you're in a google colab you can directly display it with

“a galaxy far from earth"から画像を作成し、以下のような画像になります。


まあ、“a galaxy far from earth"と言っていいかと思います。

このコードを動かすには、Hugging Faceのアカウントが必要です。あと、Stability AIのEULAの承認が必要です。