diffusion_models.models.nn.zero_module

diffusion_models.models.nn.zero_module(module)[source]

Zero out the parameters of a module and return it.