ComfyUI Extension: ComfyUI-ToDevice

Authored by Noma-Machiko

Created

Updated

0 stars

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 に移動

    インストール方法

    1. このリポジトリを ComfyUI/custom_nodes/git clone
    2. ComfyUI を再起動

    使用方法

    1. CPUで処理されるノードの直前に「ToCPU」を設置し、imageコネクタを接続してください。
       例: imageコネクタ
          →「ToCPUノード」
            →imageコネクタ
              →「CPUで実行されるノード」
    

    bash

    cd ComfyUI/custom_nodes
    git clone https://github.com/Noma-Machiko/ComfyUI-ToDevice.git