scott/kokoro
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Critical fix to 0.8.0

Browse Source
This commit is contained in:
hexgrad
2025-02-26 17:55:43 -08:00
parent efa91a8a3f
commit 3a721cce9f
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
kokoro/model.py
View File

@@ -64,7 +64,7 @@ class KModel(torch.nn.Module):
dim_out=config['n_mels'], disable_complex=disable_complex, **config['istftnet'] dim_out=config['n_mels'], disable_complex=disable_complex, **config['istftnet']
) )
if not model: if not model:
model = hf_hub_download(repo_id=repo_id, filename=MODEL_NAMES[repo_id]) model = hf_hub_download(repo_id=repo_id, filename=KModel.MODEL_NAMES[repo_id])
for key, state_dict in torch.load(model, map_location='cpu', weights_only=True).items(): for key, state_dict in torch.load(model, map_location='cpu', weights_only=True).items():
assert hasattr(self, key), key assert hasattr(self, key), key
try: try: