Google ColabでAUTOMATIC1111版web UIが動かない！？

「Google Colabでweb UIをインストールする際に警告が出る！！」
「Google ColabでStable Diffusionが使えなくなるの？」

このような場合には、この記事の内容が参考になります。
この記事では、Google ColabにおけるStable Diffusionの動作状況について説明しています。

本記事の内容

AUTOMATIC1111版web UIのインストールにおける警告表示
Stable Diffusion自体はセーフ
web UIを利用するためのオススメの方法

それでは、上記に沿って解説していきます。

AUTOMATIC1111版web UIのインストールにおける警告表示

2023年4月20日以降、Google Colab（無料版）で警告が出始めています。
警告とは、次のような警告です。

これは、AUTOMATIC1111版web UIをインストールする際に出ています。
今までなら、次のようなコマンドでAUTOMATIC1111版web UIをインストールできていました。

%pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchtext==0.14.1 torchaudio==0.13.1 torchdata==0.5.1 --extra-index-url https://download.pytorch.org/whl/cu117

!git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui
%cd /content/stable-diffusion-webui

!wget https://huggingface.co/dreamlike-art/dreamlike-photoreal-2.0/resolve/main/dreamlike-photoreal-2.0.safetensors -O /content/stable-diffusion-webui/models/Stable-diffusion/dreamlike-photoreal-2.0.safetensors

!python launch.py --share --xformers --enable-insecure-extension-access

正確には、現在（2023年4月21日）もまだインストール可能です。
アクセス可能なURLも発行されます。

そして、画像生成も普通にできます。

でも、どう考えてもこのままという訳にはいかないでしょう。
警告から始まり、そのうち利用できなくなる可能性が多いにあります。

そして、明らかにAUTOMATIC1111版web UI（stable-diffusion-webui）が狙われています。
その証拠に、次のPythonコード1行で警告が出ます。

このような状況より、脱Google Colabは考えておくべきでしょう。

以上、AUTOMATIC1111版web UIのインストールにおける警告表示を説明しました。
次は、Stable Diffusion自体はセーフについて説明します。

Stable Diffusion自体はセーフ

Google Colabにおいて、Stable Diffusionが締め出された訳ではありません。
実際、Diffusersを利用する場合は何も警告されません。

Google ColabでDiffusersを動かすには、次のコマンドだけでOK。
もちろん、GPU設定は忘れずに。

!pip install --upgrade diffusers[torch] transformers

インストールできたら、次のコードを実行します。

from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler
import torch

model_id = "stabilityai/stable-diffusion-2-1"
pipe = StableDiffusionPipeline.from_pretrained(
    model_id, 
    scheduler=EulerDiscreteScheduler.from_pretrained(
        model_id, 
        subfolder="scheduler"
    ), 
    torch_dtype=torch.float16
).to("cuda")
pipe.enable_attention_slicing()

prompt = "a photo of an astronaut riding a horse on mars"
image = pipe(prompt, height=768, width=768).images[0]
image

次のように画像が生成されます。
そして、何も警告はありません。