diffusion_models.models.nn.scale_module

diffusion_models.models.nn.scale_module(module, scale)[source]

Scale the parameters of a module and return it.