ComfyUI-BiRefNet-lite

English | 中文

Introduction

This repository packages the latest BiRefNet model as a ComfyUI node for use, supporting chunked loading on both CPU and GPU, as well as model caching features.

Features

Feature 1: Supports chunked loading on CPU and GPU

When CUDA is enabled, specify the cpu_size to load part of the model onto the CPU.

Feature 2: Model Caching

Feature 3: Multiple Cropping Methods

Supports putalpha, naive, and alpha_matting cropping methods.

Installation

Method 1:

Navigate to the node directory, ComfyUI/custom_nodes/
git clone https://github.com/rubi-du/ComfyUI-BiRefNet-lite.git
cd ComfyUI-BiRefNet-lite
pip install -r requirements.txt
Restart ComfyUI

Method 2:

Directly download the node source code package, unzip it into the custom_nodes directory, and then restart ComfyUI.

Method 3:

Install via ComfyUI-Manager by searching for "ComfyUI-BiRefNet-lite".

Usage

Example workflows are placed in ComfyUI-BiRefNet-lite/workflow.

There are two ways to load models: one is to automatically download and load a remote model, and the other is to load a local model. To load a local model, set load_local_model to true and local_model_path to the location of the local model, e.g., H:\ZhengPeng7\BiRefNet.