DiT-XL/8 — the XL backbone at patch 8.
Model Size
Parameters
pretrainedbool= FalseAccepted for signature parity with the rest of the zoo. No DiT
checkpoint ships here.
**overridesobject= {}Optional
DiTConfig field overrides.Returns
DiTModelThe denoising network.
Notes
Reference: Peebles and Xie, "Scalable Diffusion Models with Transformers", ICCV, 2023 (arXiv:2212.09748), Table 1 — 28 layers, hidden 1152, 16 heads.
Examples
>>> from lucid.models import dit_xlarge_8
>>> config = dit_xlarge_8().config
>>> config.depth, config.hidden_size, config.patch_size
(28, 1152, 8)