Auto-dispatching wrapper that loads an image-generation model.
Resolves names like "vae_gen" / "ddpm_cifar_gen" /
"ncsn_cifar_gen" to their concrete {Family}ForImageGeneration
subclass — VAE / DDPM / NCSN today, future flow models tomorrow.
Notes
All ForImageGeneration subclasses expose a generate(...)
method (via DiffusionMixin for diffusion models, or a
family-specific method for VAEs). See
GenerationOutput for the structure of the returned samples.
Examples
>>> from lucid.models import AutoModelForImageGeneration
>>> model = AutoModelForImageGeneration.from_pretrained("ddpm_cifar_gen")
>>> type(model).__name__
'DDPMForImageGeneration'