(self, control_stage_config, control_key, only_mid_control, *args, **kwargs)
| 317 | class ControlLDM(LatentDiffusion): |
| 318 | |
| 319 | def __init__(self, control_stage_config, control_key, only_mid_control, *args, **kwargs): |
| 320 | super().__init__(*args, **kwargs) |
| 321 | self.control_model = instantiate_from_config(control_stage_config) |
| 322 | self.control_key = control_key |
| 323 | self.only_mid_control = only_mid_control |
| 324 | self.control_scales = [1.0] * 13 |
| 325 | |
| 326 | @torch.no_grad() |
| 327 | def get_input(self, batch, k, bs=None, *args, **kwargs): |
nothing calls this directly
no test coverage detected