Method forward

mogen/models/transformers/mdm.py:224–227

(self, timesteps)

        )

    def forward(self, timesteps):
        output = self.time_embed(self.sequence_pos_encoder.pe[timesteps])
        output = output.permute(1, 0, 2)
        return output