Method forward

src/diffusion/unet_triplane.py:166–172

(self, x, emb)

    """

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


class TriplaneResBlock(TimestepBlock):