はじめに

ComfyUIを高火力 DOKの「ノートブック」機能を利用し、ngrokを使って接続できるようにするメモです。

今回は高火力 DOKを使用します。
高火力 DOKは、さくらインターネットが提供するコンテナ型GPUサービスです。
https://www.sakura.ad.jp/koukaryoku-dok/

前提

高火力 DOKのアカウントを持っている
- サービスを使い始めるには
ngrokのアカウントを持っている(無料プラン可)
- https://ngrok.com/

ngrokのトークンを取得

ngrokを利用するには、トークンが必要です。以下の手順で取得します。

https://dashboard.ngrok.com/ にアクセスし、サイドバーのYour Authtokenを選択します。
中央にあるYour Authtokenをメモしておきましょう。

高火力DOK

「ノートブック」の起動

ノートブックの利用方法を参考にノートブック（JupyterLab）を起動し外部から接続できるようにします。

高火力 DOKのダッシュボードにアクセスし、
サイドバーから「ノートブック」を選択したのち「起動」を選択します。

「起動」を選択し、JupyterLabが起動するまで待ちます。

「開く」を選択するとJupyterLabにアクセスできます。
※起動するまで3分程度かかる場合があるようです。

Jupyter NotebookにアクセスできたらNotebookのPython 3(ipykernel)を選択します

ComfyUIの起動

以下コードを順番に実行します。

pyngrokのインストール

!pip install pyngrok

ComfyUIをgitからDLし、必要なパッケージのインストール

!git clone https://github.com/comfyanonymous/ComfyUI.git
%cd ~/ComfyUI
!pip install -r requirements.txt

ComfyUI ManagerをgitからDLし、必要なパッケージのインストール

%cd ~/ComfyUI/custom_nodes
!git clone https://github.com/Comfy-Org/ComfyUI-Manager.git comfyui-manager
%cd ~/ComfyUI/custom_nodes/comfyui-manager
!pip install -r requirements.txt

ディレクトリの移動と成果物保存場所の変更

%cd ~/ComfyUI
!rm -rf ~/ComfyUI/output
!ln -vs /opt/artifact ~/ComfyUI/output

【任意】モデルのDL
※#(コメントアウト)を外して実行する

# !curl -Lo "/home/jovyan/ComfyUI/models/checkpoints/v1-5-pruned-emaonly-fp16.safetensors" "https://huggingface.co/Comfy-Org/stable-diffusion-v1-5-archive/resolve/main/v1-5-pruned-emaonly-fp16.safetensors"

ComfyUIの起動

ngrok_token="ngrokで取得したトークン"

from threading import Thread
from pyngrok import ngrok
import subprocess

def start_ngrok(port, token):
    ngrok.set_auth_token(token)
    url = ngrok.connect(port)
    print(f"ngrok tunnel started at: {url}")

# ngrok 起動（非同期）
ngrok_thread = Thread(target=start_ngrok, args=(8188, ngrok_token))
ngrok_thread.start()

!python main.py --preview-method auto --enable-cors-header --use-pytorch-cross-attention

起動すると
ngrok tunnel started at: NgrokTunnel: "https://xxxx.ngrok-free.app" -> "http://localhost:8188"
と表示されるので、スタートアップタスクが完了したらURLをクリックします。

URLに飛んだら「Visit site」をクリックします。

ComfyUIが開きます Yatta!!!

終わるとき

停止を忘れると従量課金されていくので忘れずに停止しましょう

高火力 DOKのダッシュボードにアクセスし、
サイドバーから「ノートブック」を選択したのち、「終了」を選択します。

ポップアップにて「終了」を選択します。

成果物について

生成した画像などの成果物は高火力 DOKのダッシュボードからダウンロードすることもできます。(期限付き)

まとめ

この記事では、高火力 DOKの「ノートブック」機能を使ってComfyUIをngrokで接続する手順を説明しました。

ここまで読んでいただきありがとうございました

参考文献

高火力 DOK ドキュメント
 ComfyUI-notebook-for-SageMaker-Studio-Lab