(self, x, condition=None)
| 413 | self.transformer.set_grad_checkpointing(set_checkpointing) |
| 414 | |
| 415 | def forward(self, x, condition=None): |
| 416 | # x [B, V*N, D] |
| 417 | x = self.layernorm1(x) |
| 418 | x = self.transformer(x, condition) |
| 419 | x = self.layernorm2(x) |
| 420 | return x |
| 421 | |
| 422 | class TransformerEncoder(TransformerBase): |
| 423 | def __init__(self, input_res, in_channels, patch_size, width, layers, heads, window_size): |
nothing calls this directly
no outgoing calls
no test coverage detected