| 54 | |
| 55 | |
| 56 | class SimpleImageConcat(AbstractLowScaleModel): |
| 57 | # no noise level conditioning |
| 58 | def __init__(self): |
| 59 | super(SimpleImageConcat, self).__init__(noise_schedule_config=None) |
| 60 | self.max_noise_level = 0 |
| 61 | |
| 62 | def forward(self, x): |
| 63 | # fix to constant noise level |
| 64 | return x, torch.zeros(x.shape[0], device=x.device).long() |
| 65 | |
| 66 | |
| 67 | class ImageConcatWithNoiseAugmentation(AbstractLowScaleModel): |
nothing calls this directly
no outgoing calls
no test coverage detected