score_sde_vp_gen(pretrained: bool = False, overrides: object = {})The VP model with its objective and the three samplers.
Model Size
Parameters
pretrainedbool= FalseNo weights are published for this family; passing
True raises.**overridesobject= {}Optional
ScoreSDEConfig field overrides.Returns
ScoreSDEForImageGenerationTrainable, and able to sample by "euler", "pc" or
"ode".
Notes
Reference: Song et al., ICLR 2021 (arXiv:2011.13456).
Examples
>>> from lucid.models import score_sde_vp_gen
>>> model = score_sde_vp_gen().eval()
>>> model.config.num_scales
1000