Extensions/ComfyUI-DiffiT
ComfyUI Extension

ComfyUI-DiffiT

Node pack that loads the official DiffiT checkpoint, creates latent noise, samples with the official pipeline path, and decodes with the selected VAE.

By Apache0ne·Created 5 months ago·Updated 5 months ago· 0
Apache0ne/ComfyUI-DiffiT
Nodes1
On cloudLocal install
CategoryNVIDIA/DiffiT
Stars0
Updated5 months ago

Nodes (1)

DiffiT Official Image

No prompts here, just 1,000 ImageNet classes and a patience requirement

NVIDIA/DiffiT
Readme

ComfyUI-DiffiT

This node pack now exposes a single node only:

  • DiffiT Official Image

It loads the official DiffiT checkpoint, creates latent noise from seed, samples with the official pipeline path, decodes with the selected VAE, and outputs IMAGE.

Class IDs ITS A LIE dont use 0

class_id is zero-based:

  • 1 = tench
  • 2 = goldfish
  • 3 = great white shark

Install

Restart ComfyUI after install.

Weights

Preferred Stability Matrix path:

Fallback ComfyUI path:

Required filenames:

  • diffit_256.safetensors
  • diffit_512.safetensors

Defaults

  • steps=250
  • scale_pow=4.0
  • cfg_scale=4.4 for diffit_256.safetensors
  • cfg_scale=1.49 for diffit_512.safetensors
  • cfg_cond=True
  • use_ddim=False
  • fp32 only

Notes

  • Image size is inferred automatically from the selected checkpoint.
  • checkpoint_path override was removed.
  • use_fp16 was removed.
  • Upstream code remains non-commercial. See LICENSE_UPSTREAM_NOTICES.md.