18 lines
540 B
Python
18 lines
540 B
Python
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
|