ComfyUI Node
FitDiT Load Model
A ComfyUI node in FitDiT with 4 inputs and 1 output.
FitDiT Load Model
- model
◄devicecuda►
◄dtypebf16►
◄offloadfalse►
◄aggressive_offloadfalse►
CategoryFitDiT
Inputs (4)
| Name | Type | Default | Description |
|---|---|---|---|
| device | COMBO | cuda | 4 options: cuda, cpu, cuda:0, cuda:1 |
| dtype | COMBO | bf16 | 2 options: bf16, fp16 |
| offload | BOOLEAN | false | — |
| aggressive_offload | BOOLEAN | false | — |
Outputs (1)
| Name | Type | Description |
|---|---|---|
| model | FITDIT_MODEL | — |