diffusion_models.modelsΒΆ Modules diffusion_models.models.fp16_util Helpers to train with 16-bit precision. diffusion_models.models.mnist_enc diffusion_models.models.multicoil diffusion_models.models.nn Various utilities for neural networks. diffusion_models.models.openai_unet diffusion_models.models.positional_encoding diffusion_models.models.vae