diffusion\_models.models.nn.scale\_module ========================================= .. currentmodule:: diffusion_models.models.nn .. autofunction:: scale_module