dit_small_4_gen(pretrained: bool = False, overrides: object = {})dit_small_4 posed as a diffusion sampler.
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
DiTForImageGenerationExposes the denoising objective and generate.
Notes
Reference: Peebles and Xie, arXiv:2212.09748, 2023.
Examples
>>> from lucid.models import dit_small_4_gen
>>> model = dit_small_4_gen()