(self, x, pos, cond)
| 537 | self.ff = FeedForwardBlock(d_model, d_ff, cond_features, dropout=dropout) |
| 538 | |
| 539 | def forward(self, x, pos, cond): |
| 540 | x = checkpoint(self.self_attn, x, pos, cond) |
| 541 | x = checkpoint(self.ff, x, cond) |
| 542 | return x |
| 543 | |
| 544 | |
| 545 | class NoAttentionTransformerLayer(nn.Module): |
nothing calls this directly
no test coverage detected