def forward()

in improved_diffusion/unet.py [0:0]

• 7 lines of code
• 3 McCabe index (conditional complexity)

    def forward(self, x, emb):
        for layer in self:
            if isinstance(layer, TimestepBlock):
                x = layer(x, emb)
            else:
                x = layer(x)
        return x