Method forward

src/models/transformer_3d.py:41–44

(self, x, c)

        )

    def forward(self, x, c):
        shift, scale = self.adaLN_modulation(c).chunk(2, dim=1)
        x = modulate(self.norm_final(x), shift, scale)
        return x

class Transformer3DModel(ModelMixin, ConfigMixin):
    _supports_gradient_checkpointing = True