ComfyUI Extension
ComfyUI_DiT [WIP]
Testbed for a/DiT(Scalable Diffusion Models with Transformers). [w/None of this code is stable, expect breaking changes if for some reason you want to use this.]
city96/ComfyUI_DiT
Nodes5
On cloudLocal install
CategoryDiT
Stars5
Updated2 years ago
Nodes (5)
DiTCheckpointLoader
The DiT loader with every dial — including one you'll never touch
DiT
DiTCheckpointLoaderSimple
The DiT loader you'll actually use (it reads the model's mind)
DiT
DiTLabelCombine
The boring node that unlocks multi-label DiT
DiT
DiTLabelSelect
Your prompt is a dropdown of 1,000 ImageNet classes
DiT
DiTSampler
The node that actually draws the picture
DiT
Readme
Old repo, use ExtraModels instead for DiT support.
ComfyUI_DiT
I needed a platform to test DiT on. None of this code is stable, expect breaking changes if for some reason you want to use this.
A lot of the code is based on the original Meta repository. I included the license for this under LICENSE-DiT.