(self, module)
| 293 | self._metadata = None |
| 294 | |
| 295 | def initialize_hook(self, module): |
| 296 | unwrapped_module = unwrap_module(module) |
| 297 | self._metadata = TransformerBlockRegistry.get(unwrapped_module.__class__) |
| 298 | return module |
| 299 | |
| 300 | @torch.compiler.disable |
| 301 | def new_forward(self, module: torch.nn.Module, *args, **kwargs): |
nothing calls this directly
no test coverage detected