()
| 177 | |
| 178 | @staticmethod |
| 179 | def get_empty_store(): |
| 180 | return {"down_cross": [], "mid_cross": [], "up_cross": [], |
| 181 | "down_self": [], "mid_self": [], "up_self": []} |
| 182 | |
| 183 | def forward(self, attn, is_cross: bool, place_in_unet: str): |
| 184 | key = f"{place_in_unet}_{'cross' if is_cross else 'self'}" |
no outgoing calls
no test coverage detected