UNET Loader Guide | Load Diffusion Model - Documentation & Example

This node has been renamed as Load Diffusion Model.

Documentation

• Class name: UNETLoader
• Category: advanced/loaders
• Output node: False

The UNETLoader node is designed for loading U-Net models by name, facilitating the use of pre-trained U-Net architectures within the system.

Input types - UNET Loader Guide | Load Diffusion Model

Parameter	Comfy dtype	Description
unet_name	COMBO[STRING]	Specifies the name of the U-Net model to be loaded. This name is used to locate the model within a predefined directory structure, enabling the dynamic loading of different U-Net models.
weight_dtype	...	🚧 fp8_e4m3fn fp9_e5m2

Output types - UNET Loader Guide | Load Diffusion Model

Parameter	Comfy dtype	Description
model	MODEL	Returns the loaded U-Net model, allowing it to be utilized for further processing or inference within the system.

Load Diffusion Model Workflow Example | UNET Loader Guide

Download UNET Loader Workflow example

1. Install the UNET models
2. Dwonload the workflow file
3. Import workflow in comfyUI
4. Chose the UNET model and run the workflow

Download FLux.1 UNET Model

FLUX.1-schnell on hugging face

File Name	Size	Link	Note
flux1-schnell.safetensors	23.8GB	Download	For lower memory usage
flux1-dev.safetensors	23.8GB	Download	If you have high VRAM and RAM.

1. Downloaded the flux1-schnell.safetensors
2. Place downloaded model files in ComfyUI/models/unet/ folder

Where you can download UNet models and how to install it?

1. You can download UNet models in folloow link:Unet Model Resource
2. After download the model files, you shou place it in /ComfyUI/models, than refresh the ComfyUI or restart it. If everything is fine, you can see the model name in the dropdown list of the UNETLoader node.