import torch
from safetensors.torch import load_file

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

if __name__ == '__main__':
    w = load_model()
    print('Absolute Value 4-bit (2\'s complement):')
    print('Input  Signed  |Abs|  Output')
    for a in range(16):
        signed = a if a < 8 else a - 16
        expected = abs(signed)
        print(f' {a:04b}   {signed:+2d}      {expected}     {expected:04b}')