(path: Path, header_size: int, info: dict)
| 68 | header_size = struct.unpack("<Q", f.read(8))[0] |
| 69 | header_json = f.read(header_size).decode("utf-8") |
| 70 | return header_size, json.loads(header_json) |
| 71 | |
| 72 | |
| 73 | def read_tensor_bytes(path: Path, header_size: int, info: dict) -> bytes: |
| 74 | start, end = info["data_offsets"] |
| 75 | with open(path, "rb") as f: |
| 76 | f.seek(8 + header_size + start) |
| 77 | return f.read(end - start) |