StableSwarmUIで手元からリモートマシンで画像生成

はじめに

サメジ部長です。AI関係の色々のために、RTX A6000を積んだ強力なサーバを借りることができました。
ところが、リモートなのでこれを使うためには自前の画像生成環境と違う環境を使う必要があるんですね。これが非常に面倒。
できることなら自分のローカルで生成を行うと自動でリモートのマシンが勝手に動いてほしい…そんな都合の良い物があったら良いのに。

ありました。StableSwarmUIです。これはComfyUIを利用していて、APIアクセスできるバックエンドとそれを操作するGUIが同梱されている画像生成AIソリューションとして利用できるんです。早い話、自分のローカルPCからリモートサーバに画像を作らせることができます。

ということでやってみた記事として紹介します。具体的な方法はほぼドキュメントに記載されていますが、リモートサーバならではのトラブルもあったので合わせて紹介します。

詰まったところ

headlessはお呼びでない？

オーダーとしてサーバ提供元から構成どうする？と聞かれたので自分はheadless環境を所望しました。ところがこれが罠だったんです。LinuxのソリューションはGUIのデスクトップがあるのを前提にしていることが結構あります。お試しで入れるとそれを前提にしていたからかエラーが起きてしまい落ちてしまいました。本来はheadlessモード用のセットアップが必要なところなのかも。

ということでheadless環境をやめて仮想ディスプレイとGNOMEをインストールしました。

sudo apt install xrdp
sudo apt -y install ubuntu-desktop task-gnome-desktop

# xrdpを起動
sudo systemctl start xrdp
# マシン再起動
sudo reboot

このGUIへはリモートデスクトップ接続などを使って画面上からアクセスしてください。

初期設定

インストール

インストール方法についてはLinuxのインストール手順を参照してマニュアルインストールしてください。
変わったところと言えばdotnet-sdk-8.0を入れなければいけないくらい。付属のインストールシェルコマンドで対処できます。例によってPython3系環境とNVIDIA GPUのドライバ、CUDA等が必要です。

以下の手順で必要なリポジトリから全て全自動で入れてくれるコマンドです。

wget https://github.com/Stability-AI/StableSwarmUI/releases/download/0.6.1-Beta/install-linux.sh -O install-linux.sh
chmod +x install-linux.sh

注意ポイント

./install-linux.sh を実行すると動くんですが、リモートマシンのデスクトップ上で行うようにしてください。
初期設定はリモートマシンのローカルからブラウザを起動して行う必要があり、そこがしばらく解決できず暗中模索してました。なにもないところからheadless状態で自動設定をしようとしてすごく時間を使いました。そんな方法はありません。
よそから設定ファイルを丸ごと持ってくるか、リモートマシン上でセットアップするかです。
本来であれば、headlessでのセットアップモードがある方が親切かな。

リモートサーバ準備

インストールに成功すると以下のようなインストーラーがリモートサーバ上のWebブラウザ上で立ち上がります。以下、インストールを進めます。

1. Agree をクリック
   

2. Just Installの状態でNextをクリック
   

3. インストールが終わるとこの画面になります。
   

4. サーバ→サーバの設定に移動し、ネットワーク：ネットワークとウェブサーバに関する設定です。の項目でポートチェンジのチェックを外してください。
   

5. ホストには0.0.0.0を指定してください。
   

6. 「Save x edited settings?」という確認が出たら保存ボタンをクリックします。

7. この設定が終わったらプロセスを終了し、 .launch-linux.shから起動し直してください。ファイルはStableSwarmUI直下にあります。

ローカルマシン設定

1. ローカルマシン上にもStableSwarmUIが必要です。こちらでのインストールは割愛します。自分はStability Matrixを経由してインストールを行いました。
2. ローカルマシンのサーバ→バックエンドに移動し、Swarm-API-Backendをクリックします。
   
3. 確認ダイアログにはOKをクリックしてください。
   
4. アドレスに http://リモートサーバのアドレス:7801/ を入力し、Saveをクリックしてください。
   
5. 確認が可能になると緑色の枠で表示されます。
   
   1. エラー時には赤枠で表示されます。設定などを見直してください。
      
6. また、リモートで画像生成をしていることを確認するため、ローカルマシンのバックエンドのComfyUIを止めておくと良いでしょう。
   
7. これで画像生成ができるようになります。注意点として、使用するモデルファイルはあらかじめローカル側とサーバ側の両方に置いておく必要があります。
   

強力なマシンを用意して快適な画像生成ライフを！！

記事は以上です。