cellflow.networks.SelfAttention.takes_attention_mask

SelfAttention.takes_attention_mask: ClassVar[bool] = False

Whether this block’s __call__ accepts an attention mask as its second positional argument. Declared as a ClassVar so it stays a pure class-level capability flag and is not turned into a flax module field (which would alter the constructor signature). _apply_modules() dispatches on this flag instead of hard-coding isinstance checks.