(self, x)
| 569 | self.layernorm2 = LayerNorm(width) |
| 570 | |
| 571 | def forward(self, x): |
| 572 | for res_log2 in range(int(np.log2(self.up_ratio))): |
| 573 | x = getattr(self, f'upsampler_{res_log2}')(x) |
| 574 | x = getattr(self, f'attention_{res_log2}')(x) |
| 575 | x = self.layernorm2(x) |
| 576 | return x |
| 577 | |
| 578 | |
| 579 |
nothing calls this directly
no outgoing calls
no test coverage detected