ComfyUI Extension: ComfyUI-ToDevice
A simple collection of device conversion nodes for ComfyUI. These nodes move an image node running on the GPU over to the CPU. They also support moving from CPU to GPU, but in that case, please remove the CPU option from the ComfyUI startup options.
Custom Nodes (0)
README
ComfyUI-ToDevice
ComfyUI 用のシンプルなデバイス変換ノード集です。
GPUで実行されているimageノードを、CPUへ移動させます。
CPUからGPUの移動も対応していますが、ComfyUIの起動オプションからCPUを消去してください。
ノード一覧
- ToCPU: 入力画像テンソルを CPU に移動
- ToGPU: 入力画像テンソルを GPU に移動
インストール方法
- このリポジトリを
ComfyUI/custom_nodes/
にgit clone
- ComfyUI を再起動
使用方法
- CPUで処理されるノードの直前に「ToCPU」を設置し、imageコネクタを接続してください。
例: imageコネクタ
→「ToCPUノード」
→imageコネクタ
→「CPUで実行されるノード」
bash
cd ComfyUI/custom_nodes
git clone https://github.com/Noma-Machiko/ComfyUI-ToDevice.git