dit_small_2_gen

→DiTForImageGeneration

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

dit_small_2 posed as a diffusion sampler.

pretrainedbool= False

No DiT checkpoint was published for this configuration; True raises rather than returning a randomly initialised model.

**overridesobject= {}

Optional DiTConfig field overrides.

DiTForImageGeneration

Exposes the denoising objective and generate.

Reference: Peebles and Xie, arXiv:2212.09748, 2023.

>>> from lucid.models import dit_small_2_gen
>>> model = dit_small_2_gen()