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,…

By Noma-Machiko·Created 11 months ago·Updated 11 months ago· 1

Noma-Machiko/ComfyUI-ToDevice

Nodes2

On cloudLocal install

Categoryutils

Stars1

Updated11 months ago

Nodes (2)

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