()
| 8 | |
| 9 | @staticmethod |
| 10 | def get_empty_store(): |
| 11 | return {8: [], 16: [], 32: [], 64: []} |
| 12 | # return {"down_cross": [], "mid_cross": [], "up_cross": [], |
| 13 | # "down_self": [], "mid_self": [], "up_self": []} |
| 14 | |
| 15 | # def forward(self, attn, is_cross: bool = True, place_in_unet: str = None): |
| 16 | def __call__(self, attn, is_cross: bool = True, place_in_unet: str = None): |
no outgoing calls
no test coverage detected