From 3a721cce9f03d73bf073aede362b3506823be1b1 Mon Sep 17 00:00:00 2001 From: hexgrad Date: Wed, 26 Feb 2025 17:55:43 -0800 Subject: [PATCH] Critical fix to 0.8.0 --- kokoro/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/kokoro/model.py b/kokoro/model.py index 08ffe8f..51e835b 100644 --- a/kokoro/model.py +++ b/kokoro/model.py @@ -64,7 +64,7 @@ class KModel(torch.nn.Module): dim_out=config['n_mels'], disable_complex=disable_complex, **config['istftnet'] ) 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(): assert hasattr(self, key), key try: