(self, hidden_states)
| 412 | return output |
| 413 | |
| 414 | def forward(self, hidden_states): |
| 415 | if self.use_conv: |
| 416 | downsample_input = self._downsample_2d(hidden_states, weight=self.Conv2d_0.weight, kernel=self.fir_kernel) |
| 417 | hidden_states = downsample_input + self.Conv2d_0.bias.reshape(1, -1, 1, 1) |
| 418 | else: |
| 419 | hidden_states = self._downsample_2d(hidden_states, kernel=self.fir_kernel, factor=2) |
| 420 | |
| 421 | return hidden_states |
| 422 | |
| 423 | |
| 424 | # downsample/upsample layer used in k-upscaler, might be able to use FirDownsample2D/DirUpsample2D instead |
nothing calls this directly
no test coverage detected