diffusion\_models.models.nn.checkpoint ====================================== .. currentmodule:: diffusion_models.models.nn .. autofunction:: checkpoint