| 271 | |
| 272 | @dataclass(frozen=True) |
| 273 | class TargetCacheSampleBytes: |
| 274 | sample_id: int |
| 275 | seq_len: int |
| 276 | input_ids: bytes |
| 277 | attention_mask: bytes |
| 278 | loss_mask: bytes |
| 279 | target_hidden_states: bytes |
| 280 | target_last_hidden_states: bytes |
| 281 | |
| 282 | |
| 283 | def build_target_cache_sample_bytes( |
no outgoing calls
no test coverage detected