| 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 |
|
|
