How to access header with python

[justinchuby]

Is there a way to access the header in Python to know the offsets of each tensor data?

[justinchuby]

I am now using this snippet:

def read_safetensors_header(file: BytesIO) -> dict[str, dict[str, Any]]:
    """Read the header of a safetensors file.

    Args:
        file: The safetensors file to read.

    Returns:
        The header of the safetensors file.
    """
    header_size = struct.unpack_from("i", file.read(8))[0]
    header = file.read(header_size)
    return json.loads(header.decode("utf-8"))

Please let me know if there are better ways.