Method forward

models/transformer/wan/modules/tm2m_model.py:548–553

(self, x)

        self.register_buffer('pe', pe)

    def forward(self, x):
        # x: [B, T, C]
        x = x.permute(1, 0, 2)
        x = x + self.pe[:x.size(0), :]
        x = x.permute(1, 0, 2)
        return x

class WanModelTM2M(nn.Module):