Method
__init__
(self,
concat_keys=("mask", "masked_image"),
masked_image_key="masked_image",
*args, **kwargs
)
Source from the content-addressed store, hash-verified
| 1647 | """ |
| 1648 | |
| 1649 | def __init__(self, |
| 1650 | concat_keys=("mask", "masked_image"), |
| 1651 | masked_image_key="masked_image", |
| 1652 | *args, **kwargs |
| 1653 | ): |
| 1654 | super().__init__(concat_keys, *args, **kwargs) |
| 1655 | self.masked_image_key = masked_image_key |
| 1656 | assert self.masked_image_key in concat_keys |
| 1657 | |
| 1658 | @torch.no_grad() |
| 1659 | def get_input(self, batch, k, cond_key=None, bs=None, return_first_stage_outputs=False): |
Callers
nothing calls this directly
Tested by
no test coverage detected