Function zero_module

utils/unet_blocks.py:494–498

(module)

# Animatediff_motion_module
def zero_module(module):
    # Zero out the parameters of a module and return it.
    for p in module.parameters():
        p.detach().zero_()
    return module


@dataclass