def forward()

in glide_text2im/unet.py [0:0]

• 9 lines of code
• 4 McCabe index (conditional complexity)

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