(self, diff_model_config, conditioning_key)
| 1319 | |
| 1320 | class DiffusionWrapper(pl.LightningModule): |
| 1321 | def __init__(self, diff_model_config, conditioning_key): |
| 1322 | super().__init__() |
| 1323 | self.sequential_cross_attn = diff_model_config.pop("sequential_crossattn", False) |
| 1324 | self.diffusion_model = instantiate_from_config(diff_model_config) |
| 1325 | self.conditioning_key = conditioning_key |
| 1326 | assert self.conditioning_key in [None, 'concat', 'crossattn', 'hybrid', 'adm', 'hybrid-adm', 'crossattn-adm'] |
| 1327 | |
| 1328 | def forward(self, x, t, c_concat: list = None, c_crossattn: list = None, c_adm=None): |
| 1329 | if self.conditioning_key is None: |
nothing calls this directly
no test coverage detected