1 クリックで立ち上がる自分だけの環境で、お猫様の画像生成をやってみた。- 拡張編 -

Stable Diffusion Web UI の Extension タブから、有志が開発した拡張機能をインストールすることができます。

画像では、入力したプロンプトを自動で補完 (オートコンプリート) してくれる拡張機能を導入した例になります。

次々にプロンプトを補完してくれるため非常に便利です。

Amazon EC2 や EC2 にマウントした Amazon S3 の中にあるファイルに、ブラウザ経由でアクセスすることができます。

こちらの画像は、Stable Diffusion Web UI で生成した画像が保存されているディレクトリを参照している例です。

生成した画像を簡単に管理できることに加え、ローカル PC からファイルをアップロードすることもできるため、ローカルから直接モデルを EC2 にアップロードして利用することもできます。

ブラウザ経由でモデルのトレーニングを行うことができます。

例えば、「我が家の猫が一番可愛い ! 我が家の猫を生成するモデルを作りたい !」と思い立ったとしましょう。

この機能があれば、数枚の画像を用意し、Stable Diffusion の画像モデルのファインチューニングを行うことで、「我が家の猫」を生成するモデルを作成することができます !

次に、起動する EC2 にパブリック IP アドレスを自動的に付与するための設定を行います。

サブネットのコンソール画面にて、上記手順で控えたサブネットを選択し、「アクション」→「サブネットの設定を編集」を押します。

「パブリック IPv4 アドレスの自動割り当てを有効化」にチェックを入れて、「保存」を押しましょう。

IP アドレスの自動割り当て設定を有効化することで、このサブネット内に作成する Stable Diffusion Web UI 用の EC2 インスタンスに自動的にパブリック IPv4 アドレスを割り当てることができます。

※ 既存の VPC を利用する場合など、すでにこの設定が有効化されている場合は作業は不要です。

今回の構成では、EC2 に S3 バケットをマウントして、ストレージとして利用できるようにします。

マネジメントコンソールの検索窓から「S3」を検索し、表示された「S3」を押します。

「バケット」を選択し、「バケットを作成」を押します。

任意のバケット名を設定します。バケット名はグローバルで一意である必要があるので、被らない名前を設定しましょう。例えば、sd-webui-bucket-[名前]-[yyyymmdd] といったネーミングだと被りにくそうです。今回は sd-webui-bucket-kobayashi にしてみました。

AWS リージョンは、EC2 を起動するリージョンと同じものを設定します。

CloudFormation テンプレートはデフォルトで us-east-1 用になっているため、今回は「us-east-1」を選択しています。もしも us-east-1 以外のリージョンで VPC を構築されている場合は、こちらを VPC に対応したリージョンに変更してください。

入力できたら画面をスクロールして、「バケットを作成」を押します。

作成したバケット名は CloudFormation の設定に使うため、控えておきましょう。

マネジメントコンソールの検索窓に「IAM」と入力し、表示された「IAM」を押します。

IAM コンソールのナビゲーションペインで、「ロール」を選択し、表示された「ロールを作成」を押します。

信頼されたエンティティタイプとして「AWS のサービス」を選択します。

ユースケースは「EC2」を選び、「次へ」を押します。

検索欄に AmazonSSMManagedInstanceCore と入力し、表示されたポリシーにチェックを入れます。

次に、S3 に対するアクセス許可も設定しましょう。

AmazonS3FullAccess と検索し、表示されたポリシーにチェックを入れて、「許可を追加」を押します。

※ この記事では簡単のために、AmazonS3FullAccess 権限を付与していますが、実際に運用される際は、バケット単位で許可を与えるなど、最小限のアクセス許可のみ付与することが推奨されています。ポリシーの書き方については こちら をご参照ください。

任意のロール名と説明を入力します。ここでは sd-ec2-role としています。

先ほど選択した 2 つのポリシーが設定されていることを確認し、画面をスクロールして、「ロールを作成」を押します。

ロールの一覧画面に戻ったら、検索欄に先ほど作成したロール名を入力して絞り込みます。

ロール名が表示されれば OK です。このロール名は CloudFormation テンプレートで利用するため、控えておいてください。

画面左から「ユーザー」を選択し、「ユーザーの作成」を押します。

任意のユーザー名を設定します。ここでは、sd-ec2-kobayashi としました。

入力できたら「次へ」を押しましょう。

この IAM ユーザーに割り当てる権限を設定していきます。
「ポリシーを直接アタッチする」を選び、AmazonSSMFullAccess ポリシーを割り当て、「次へ」を押します。

※ 今回は簡単のため AmazonSSMFullAccess 権限を付与していますが、実際の運用ではアクセス可能な EC2 インスタンスを制限するなど、最小限のアクセス許可のみ付与することが推奨されています。ポリシーの書き方については こちら をご参照ください。

確認画面が表示されたら、「ユーザーの作成」を押します。

ユーザーの一覧画面に戻るので、作成されたユーザー名を押しましょう。

このユーザー ID を利用するための認証情報を作成します。

「セキュリティ認証情報」タブから、「アクセスキーを作成」を押しましょう。

ユースケースを選択します。

「コマンドラインインターフェース(CLI)」を選択し、「上位機のレコメンデーションを理解し、アクセスキーを作成します」にチェックを入れて、「次へ」を押します。

この認証情報をわかりやすく判別するためにタグを設定することも可能です。

「アクセスキーを作成」を押します。

こちらがユーザー ID の認証情報です。
アクセスキーと、シークレットアクセスキーの値を控えておきましょう。

※ この 2 つの情報があれば、誰でも事前に設定した権限を利用して AWS の操作を行うことができてしまうため、取り扱いにはご注意ください。

sd-webui-kohya-private-s3.yaml の左にある「Launch Stack」ボタンを押しましょう。

「Launch Stack」ボタンを押すと sd-webui-kohya-private-s3.yaml テンプレートが自動的に読みこまれ、CloudFormation スタックの詳細を設定する画面が開きます。

ここでは、以下の値を設定します。

• EC2InstanceProfileName : 作成した IAM ロール名
• S3BucketName : 作成した S3 バケット名
• SubnetId : EC2 を起動するサブネットの ID

設定できたら、「スタックの作成」を押しましょう。

スタックの作成が開始されると、CloudFormation の「イベント」タブにリソースの作成状況が表示されます。

数分ほど待って、論理 ID 欄が sd-webui (スタック名) のレコードのステータス欄が CREATE_COMPLETE ステータスに変わった後、20 分ほど待つと環境構築が完了します。

※ CREATE_COMPLETE が表示された時点で EC2 インスタンスの作成は完了していますが、EC2 インスタンスの内部で Stable Diffusion Web UI 等のインストール等が実行されるため、ブラウザから接続できる準備が整うまで 20 分程度かかります。

20 分ほど待ったら、EC2 への各種インストール状況の進捗を確認してみましょう。

Cloud Formation の「出力」タブに表示された Instance ID をクリックし、インスタンスの詳細画面に遷移します。

ステータス欄の表示が「2/2 のチェックに合格しました」と表示されているかを確認します。

「初期化しています」と表示されている場合は、表示が変わるまで待ちましょう。

画面右上の更新ボタンを押すと、画面を更新することができます。

インスタンスを選択して、「接続」を押します。

「セッションマネージャー」を選択し、「接続」を押します。

セッションマネージャー経由で EC2 インスタンスに接続することができました。

EC2 のコンソールを開き、画面左から「セキュリティグループ」を選択し、「セキュリティグループを作成」を押します。

クリックすると拡大します

セキュリティグループの情報を設定します。「セキュリティグループ名」、「説明」には任意の値を入力してください。ここでは sd-ec2-sg としています。

VPC は EC2 を起動している VPC を選択しましょう。

クリックすると拡大します

インバウンドルール欄には何も設定されておらず、アウトバウンドルール側はすべてのトラフィックを通すように設定されていることを確認し、「セキュリティグループを作成」を押します。

クリックすると拡大します

次に、作成したセキュリティグループを EC2 インスタンスにアタッチします。

EC2 インスタンスを選択し、「アクション」→「セキュリティ」→「セキュリティグループを変更」を押します。

クリックすると拡大します

「削除」を押して、デフォルトのセキュリティグループを削除します。

クリックすると拡大します

検索欄から、先ほど作成したセキュリティグループを選択し、「セキュリティグループを追加」を押します。

関連づけられたセキュリティグループ欄に選択したセキュリティグループが表示されたことを確認し、「保存」を押します。

クリックすると拡大します

ブラウザを起動し、localhost:7860 にアクセスしましょう。

このような画面が表示されたら、Stable Diffusion Web UI へのアクセスは成功です。

クリックすると拡大します

試しに前回と同じくネコチャンの画像を生成してみましょう。

Prompt に a cat と入力し、「Generate」を押します。

クリックすると拡大します

今回構築した環境では、Stable Diffusion Web UI 上から拡張機能のインストールが可能です。

「Extentions」タブを開き、「Install from URL」を押しましょう。

クリックすると拡大します

お試しとして、a1111-sd-webui-tagcomplete をインストールしてみます。こちらは、入力したプロンプトを自動で補完（オートコンプリート）してくれる拡張機能です。

「URL for extension's git repository」欄に Github リポジトリの URL を入力し、「install」を押します。

クリックすると拡大します

「Installed」タブに切り替え、下部の拡張機能一覧に先ほどインストールした「a1111-sd-webui-tagcomplete」が表示されていることを確認します。

「Apply and restart UI」を押して、拡張機能を有効化しましょう。

クリックすると拡大します

「txt2img」タブに切り替え、プロンプトを入力してみます。
すると、プロンプトを入力した際に補完候補が出るようになりました !

Stable Diffusion Web UI には有志が開発した拡張機能が数多くあるため、ぜひ色々と試してみてください。

クリックすると拡大します

次は localhost:7861 にアクセスしてみましょう。
こちらの画面が表示されれば、Kohya's GUI の起動ができています。

クリックすると拡大します

次は localhost:8080 にアクセスしてみましょう。
このような画面が表示されたら ID / Password にそれぞれ admin と入力し、「Login」を押します。

クリックすると拡大します

このような画面が表示されれば、File Browser の起動ができています。

File Browser では EC2 上のディレクトリや、EC2 にマウントされた S3 にブラウザベースでアクセスすることができます。

クリックすると拡大します

例えば、「stable-diffusion-webui」→「outputs」→「txt2img-images」→「日付」フォルダにアクセスすると、生成した画像を確認することができます。ダウンロードやアップロードもブラウザから可能です。

クリックすると拡大します

対象の EC2 インスタンスを選択し、「インスタンスの状態」→「インスタンスの停止」を押すことで、EC2 インスタンスを停止することができます。

再度 Stable Diffusion Web UI を触りたいときは、「インスタンスの状態」→「インスタンスを開始」を押すことで、インスタンスを起動することが可能です。

※ このとき「インスタンスを終了」を押すと、EC2 インスタンスが削除されてしまい、生成した画像も一緒に消えてしまうためご注意ください。

クリックすると拡大します

また、環境が不要になった際は、マネジメントコンソールの CloudFormation の画面から「削除」を押すことで EC2 を含めたリソースを削除することが可能です。

※ S3 バケットや IAM ユーザーなど、CloudFormation の外部で作成したリソースは個別に削除する必要があります。

クリックすると拡大します