(self, x, timesteps, low_res=None, **kwargs)
| 540 | return super().forward(x, timesteps, **kwargs) |
| 541 | |
| 542 | def get_feature_vectors(self, x, timesteps, low_res=None, **kwargs): |
| 543 | _, new_height, new_width, _ = x.shape |
| 544 | upsampled = F.interpolate(low_res, (new_height, new_width), mode="bilinear") |
| 545 | x = th.cat([x, upsampled], dim=1) |
| 546 | return super().get_feature_vectors(x, timesteps, **kwargs) |
| 547 |
nothing calls this directly
no test coverage detected