(self, transformer_block, mask, token_idx, do_classifier_free_guidance, scale_factors)
| 99 | |
| 100 | class PAIntAAttnProcessor: |
| 101 | def __init__(self, transformer_block, mask, token_idx, do_classifier_free_guidance, scale_factors): |
| 102 | self.transformer_block = transformer_block # Stores the parent transformer block. |
| 103 | self.mask = mask |
| 104 | self.scale_factors = scale_factors |
| 105 | self.do_classifier_free_guidance = do_classifier_free_guidance |
| 106 | self.token_idx = token_idx |
| 107 | self.shape = mask.shape[2:] |
| 108 | self.mask_resoltuion = mask.shape[-1] * mask.shape[-2] # 64 x 64 |
| 109 | self.default_processor = AttnProcessor() |
| 110 | |
| 111 | def __call__( |
| 112 | self, |
nothing calls this directly
no test coverage detected