from hashlib import sha256
from pathlib import Path
import torch

path = Path(__file__).parent.parent / 'kokoro-v0_19.pth'
assert path.exists(), f'No model pth found at {path}'

net = torch.load(path, map_location='cpu', weights_only=True)['net']
for a in net:
    for b in net[a]:
        net[a][b] = net[a][b].half()

torch.save(dict(net=net), 'kokoro-v0_19-half.pth')
with open('kokoro-v0_19-half.pth', 'rb') as rb:
    h = sha256(rb.read()).hexdigest()

assert h == '70cbf37f84610967f2ca72dadb95456fdd8b6c72cdd6dc7372c50f525889ff0c', h