def forward()

in janus/janusflow/models/uvit.py [0:0]

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

    def forward(self, x, emb, recompute=False):
        for res_block in self.res_blocks:
            x = res_block(x, emb)

        if self.downsample is not None:
            x = self.downsample(x)

        if self.upsample is not None:
            x = self.upsample(x)

        return x