(self, attn, is_cross: bool, place_in_unet: str)
| 144 | |
| 145 | @abc.abstractmethod |
| 146 | def forward(self, attn, is_cross: bool, place_in_unet: str): |
| 147 | raise NotImplementedError |
| 148 | |
| 149 | def __call__(self, attn, is_cross: bool, place_in_unet: str): |
| 150 | if self.cur_att_layer >= self.num_uncond_att_layers: |