| 180 | |
| 181 | @dataclass |
| 182 | class BlockModelInput: |
| 183 | doc_input_ids: torch.Tensor |
| 184 | doc_attention_mask: torch.Tensor |
| 185 | doc_ids: torch.Tensor |
| 186 | position_ids: torch.Tensor |
| 187 | num_chunks: int |
| 188 | chunk_sizes: List[int] |
| 189 | |
| 190 | @dataclass |
| 191 | class BlockDesc: |
nothing calls this directly
no outgoing calls
no test coverage detected