dit_small_2

→DiTModel

dit_small_2(pretrained: bool = False, overrides: object = {})

DiT-S/2 — the S backbone at patch 2.

pretrainedbool= False

Accepted for signature parity with the rest of the zoo. No DiT checkpoint ships here.

**overridesobject= {}

Optional DiTConfig field overrides.

DiTModel

The denoising network.

Reference: Peebles and Xie, "Scalable Diffusion Models with Transformers", ICCV, 2023 (arXiv:2212.09748), Table 1 — 12 layers, hidden 384, 6 heads.

>>> from lucid.models import dit_small_2
>>> config = dit_small_2().config
>>> config.depth, config.hidden_size, config.patch_size
(12, 384, 2)