(self, *args, low_scale_config, low_scale_key="LR", noise_level_key=None, **kwargs)
| 1361 | |
| 1362 | class LatentUpscaleDiffusion(LatentDiffusion): |
| 1363 | def __init__(self, *args, low_scale_config, low_scale_key="LR", noise_level_key=None, **kwargs): |
| 1364 | super().__init__(*args, **kwargs) |
| 1365 | # assumes that neither the cond_stage nor the low_scale_model contain trainable params |
| 1366 | assert not self.cond_stage_trainable |
| 1367 | self.instantiate_low_stage(low_scale_config) |
| 1368 | self.low_scale_key = low_scale_key |
| 1369 | self.noise_level_key = noise_level_key |
| 1370 | |
| 1371 | def instantiate_low_stage(self, config): |
| 1372 | model = instantiate_from_config(config) |
nothing calls this directly
no test coverage detected