import torch
from safetensors.torch import load_file

def load_model(path='model.safetensors'):
    return load_file(path)

def crc4_ref(d3, d2, d1, d0):
    """CRC-4 with polynomial x^4+x+1."""
    crc0 = d3 ^ d0
    crc1 = d2 ^ d1 ^ d0
    crc2 = d3 ^ d1
    crc3 = d3 ^ d2
    return crc3, crc2, crc1, crc0

if __name__ == '__main__':
    print('CRC-4 Generator (polynomial x^4+x+1)')
    print('Data -> CRC')
    for i in range(16):
        d3, d2, d1, d0 = (i >> 3) & 1, (i >> 2) & 1, (i >> 1) & 1, i & 1
        c3, c2, c1, c0 = crc4_ref(d3, d2, d1, d0)
        print(f'  {d3}{d2}{d1}{d0} -> {c3}{c2}{c1}{c0}')