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

みなさん、こんにちは ! VTuber 大好きソリューションアーキテクトの小林 D です。

 

突然ですが、みなさんは Stable Diffusion という画像生成 AI を使ったことはありますか ? Stable Diffusion は英国の Stability.AI 社が開発した、入力したテキスト (プロンプト) から画像を生成することができる画像生成 AI です。

Stable Diffusion はオープンソースのモデルとなっており、Amazon SageMaker JumpStart 上でも簡単に動かすことができます。

以下は SageMaker JumpStart で Stable Diffusion を利用する AWS Blog の記事です。

• Stable Diffusion で画像の部分的な差し替えを行う環境を、 Amazon SageMaker JumpStart で簡単に構築する

• たった数枚の画像で Stable Diffusion をファインチューニングできる効率的な Amazon SageMaker JumpStart の使い方

このように、SageMaker JumpStart を使うことで、SageMaker や Jupyter Notebook に慣れた方は簡単に Stable Diffusion のをデプロイし、システムに組み込むことができます。

一方で、システムに組み込むのではなく、業務の補助ツールとして Stable Diffusion を利用したい、というニーズもあります。

実際に、お客様に対して SageMaker JumpStart の画面をお見せして画像生成のデモをした際、デザイナーの方から「自分でも弄ってみたいけど、Python や Jupyter Notebook に馴染みがないので難しそう」「GUI で細かくパラメータ調整をしたい」というお声をいただくことがありました。

そこで利用できるのが AUTOMATIC1111 氏が公開している Stable Diffusion Web UI です。

Stable Diffusion Web UI ではブラウザベースの UI で Stable Diffusion を利用することができるツールとなっています。そのため、コードを書くことなく、様々なパラメータを直感的に操作することが可能です。

ということで、この記事では Amazon EC2 を利用して、AWS 上に Stable Diffusion Web UI を利用できる環境を作っていきたいと思います !

X ポスト » | Facebook シェア » | はてブ »

こちらのリンク から「Running On-Demand G and VT instances」 のページを開きます。

ページが開いたら、「リージョン」を選択 を押してクォータを確認したいリージョンを選択します。

※この記事の手順ではバージニア北部リージョンを利用します

適用されたクォータ値の値を確認し、値が 4 未満であれば「クォータの引き上げをリクエスト」を押して、クォータを 4 以上に引き上げます。

※ なぜ 4 以上に引き上げるかというと、Running On-Demand G and VT instances の設定値はインスタンスの数ではなく vCPU の数になっているからです。g4dn.xlarge インスタンスが利用する vCPU 数は 4 です。そのため、最低 4 vCPU で申請を行いましょう。

検索欄に「VPC」と入力し、表示された VPC を押します。

「 VPC を作成」を押します。

VPC の作成画面が表示されるので、「VPC など」を選択します。

「名前タグの自動生成」欄に、任意の名前を入力しましょう。ここでは stable-diffusion とします。

画面右側のプレビュー欄を見ると、「 名前タグの自動生成」欄に入力した値が VPC 内のリソースに付与されており、識別がしやすくなりました。

その他の値はデフォルトのままで問題ありません。

画面を一番下までスクロールし、「VPC を作成」を押します。

VPC に加えて、インターネットに接続するための経路であるインターネットゲートウェイや、ルートテーブルの設定などが自動的に行われます。

「成功」が表示されたら、右下の「VPC を表示」を押しましょう。

表示された VPC ID の値を控えておきます。

※ Stable Diffusion Web UI 環境を立ち上げる際、VPC を指定するために利用します。

画面下の Resource map 欄に表示されているサブネットを確認します。

サブネットは、VPC を小さく分割したネットワークセグメントのことです。

名前の中に public と入ったサブネットと、private が入ったサブネットが 2 つずつ作成されています。前者はパブリックサブネットと呼ばれており、インターネットへアクセスする経路があるサブネットのことを指します。後者はプライベートサブネットと呼ばれており、インターネットへアクセスする経路がない、閉じたサブネットのことを指します。

今回はパブリックにアクセス可能な環境を作成するため、EC2 インスタンスが起動するサブネットはパブリックサブネットを指定します。

パブリックサブネットの ID を確認するため、「stable-diffusion-subnet-public1...」という名前のサブネットをクリックして、サブネットの詳細画面を開きましょう。

「 サブネット ID」の値を控えておきましょう。

CloudFormation テンプレートをダウンロードし、マネジメントコンソールから読み込む方法もあるのですが、「Launch Stack」ボタンを押すことで、直接 CloudFormation のスタック作成ページに飛ぶことができます。

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

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

起動する EC2 の設定や、Stable Diffusion Web UI のバージョン情報などはデフォルトの値が設定されているため、SubnetId と VpcId の値に、先ほど控えた「サブネット ID の値」、「VPC ID の値」をそれぞれ設定し、「スタックの作成」を押しましょう。

ここで指定した VPC のサブネットに、EC2 インスタンスが起動します。

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

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

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

「出力」タブを確認すると、作成されたリソースの情報を確認することができます。

• InstanceID : EC2 インスタンスの ID です。
• Publicip : EC2 インスタンスに割り当てられた IP アドレスです。

Publicip の IP アドレスの値はこのあと Stable Diffusion Web UI に接続する際に利用するので、控えておきましょう。

※ もしここで、このようなエラーが出て EC2 インスタンスの作成に失敗する場合は、GPU インスタンスのクォータ (起動台数の上限) に抵触しています。 1. 事前準備 を参考にクォータの引き上げを行ってください。

クォータの引き上げ後、スタックを再度作成する際には、作成に失敗したスタックは削除しておきましょう。

「削除」ボタンからスタックの削除を行うことができます。

スタックの削除後、再度 Github リポジトリの「Launch Stack」ボタンからスタックを作成しましょう。

CloudFormation によるリソースの作成完了から 10 分程度経ったら、Stable Diffusion Web UI の実行状況を確認してみましょう。検索欄に「 EC2」と入力し、表示された「 EC2」を押します。

EC2 ダッシュボードが開くので。「 インスタンス (実行中)」を押します。

CloudFormation によって作成された sd-web-ui-cf インスタンスが実行中になっています。

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

インスタンスの接続画面が開くので「EC2 Instance Connect」タブが選択されていることを確認し、「EC2 Instance Connect を使用して接続する」を選びます。

右下の「接続」ボタンを押して、ブラウザ経由で EC2 インスタンスに接続しましょう。

EC2 Instance Connect 経由で EC2 インスタンスに接続できました。

このような画面が表示されたら、Stable Diffusion Web UI へのアクセスに成功しています !

試しにお猫様の画像を生成してみましょう。Prompt に a cat と入力し、「Generate」を押します。

かわいいお猫様が出力されました !

EC2 のインスタンス一覧を開き、sd-web-ui-cf インスタンスを選択します。

画面下に表示される「セキュリティ」タブを選択し、「セキュリティグループ」欄に表示されているセキュリティグループ ID を押しましょう。

「 インバウンドのルールを編集」を押します。

現在はポート 22 と 7860 に対して、0.0.0.0/0 からのアクセスが許可されています。0.0.0.0/0 はすべての IP アドレスを意味しているため、どこからでもアクセスができる状態になっています。

こちらを自分の IP アドレスだけに絞りましょう。

それぞれのレコードの 0.0.0.0/0 の隣の × を押します。

「カスタム」→「マイ IP」を選択します。

現在操作を行っている環境の IP アドレスが自動的に入力されます。

このように、自分の IP アドレスが設定されれば OK です。

「ルールを保存」ボタンを押して、設定を保存しましょう。

変更後のインバウンドルールのソース欄の値が、自身の IP アドレスの値になっていれば変更は完了です。

ここで指定した IP アドレス 以外からは Stable Diffusion Web UI にアクセスできないようになりました !

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

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

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

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

小林 大樹
アマゾン ウェブ サービス ジャパン合同会社
ソリューションアーキテクト

VTuber をこよなく愛するソリューションアーキテクト。普段は業種業界問わず、様々なお客様のプロダクト開発をサポートさせていただいております。悩みは推し VTuber が増えすぎて視聴時間を確保できないことです。