Method
__init__
(self, block, block_idx, global_block_idx, offloader)
Source from the content-addressed store, hash-verified
| 448 | |
| 449 | class SingleTransformerWrapper(nn.Module): |
| 450 | def __init__(self, block, block_idx, global_block_idx, offloader): |
| 451 | super().__init__() |
| 452 | self.block = block |
| 453 | self.block_idx = block_idx |
| 454 | self.global_block_idx = global_block_idx |
| 455 | self.offloader = offloader |
| 456 | |
| 457 | @torch.autocast('cuda', dtype=AUTOCAST_DTYPE) |
| 458 | def forward(self, inputs): |
Callers
nothing calls this directly
Tested by
no test coverage detected