(self, block, stg_flag: STGFlag, idx: int)
| 170 | """Wraps transformer blocks to be able to skip attention layers.""" |
| 171 | |
| 172 | def __init__(self, block, stg_flag: STGFlag, idx: int): |
| 173 | self.flag = stg_flag |
| 174 | self.idx = idx |
| 175 | self.block = block |
| 176 | |
| 177 | def __call__(self, args, extra_args): |
| 178 | context_manager = contextlib.nullcontext() |
nothing calls this directly
no outgoing calls
no test coverage detected