Function zero_module

diffusers/src/diffusers/models/controlnet_sparsectrl.py:785–788

(module: nn.Module)

# Copied from diffusers.models.controlnet.zero_module
def zero_module(module: nn.Module) -> nn.Module:
    for p in module.parameters():
        nn.init.zeros_(p)
    return module