高火力 DOKの「ノートブック」機能を使ってComfyUIをngrokで接続する

Posted at 2025-04-01

はじめに

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