Function zero_module

src/diffusers/models/brushnet.py:928–931

(module)

def zero_module(module):
    for p in module.parameters():
        nn.init.zeros_(p)
    return module