| 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}') | |