(self, x)
| 504 | self.register_buffer('pe', pe) |
| 505 | |
| 506 | def forward(self, x): |
| 507 | # x: [B, T, C] |
| 508 | x = x.permute(1, 0, 2) |
| 509 | x = x + self.pe[:x.size(0), :] |
| 510 | x = x.permute(1, 0, 2) |
| 511 | return x |
| 512 | |
| 513 | class WanModelT2M(nn.Module): |
| 514 |
nothing calls this directly
no outgoing calls
no test coverage detected