Method forward

k_diffusion/models/modules.py:556–559

(self, x, *args, **kwargs)

class Level(nn.ModuleList):
    def forward(self, x, *args, **kwargs):
        for layer in self:
            x = layer(x, *args, **kwargs)
        return x


# Mapping network