A small modification of the ComfyUI_bitsandbytes_NF4 extension that allows loading Diffusion Models separately from text encoders and VAE.

This may be useful for users who have already downloaded T5, CLIP and VAE to save disk space.

Extension adds UNETLoaderNF4 node (in advanced/loaders category).

NF4 Diffusion Models can be obtained on HuggingFace. Download Diffusion Model and place it in ComfyUI/models/unet.

Diffusion Models available:

• flux1-dev-bnb-nf4-unet
• flux1-dev-bnb-nf4-v2-unet
• flux1-schnell-bnb-nf4-unet
• flux1-schnell-bnb-nf4-v2-unet

Requires installing bitsandbytes.

Installation:

• Open terminal in ComfyUI/custom_nodes and use this command to clone repo:

  git clone https://github.com/DenkingOfficial/ComfyUI_UNet_bitsandbytes_NF4.git

Workflow

Original README:

Now on the manager for easy installation. Make sure to select Channel:dev in the ComfyUI manager menu or install via git url.

A quickly written custom node that uses code from Forge to support the nf4 flux dev checkpoint and nf4 flux schnell checkpoint.

Requires installing bitsandbytes.

Make sure your ComfyUI is updated.

The node is: CheckpointLoaderNF4, just plug it in your flux workflow instead of the regular one.