From e409c4a19b911f203de69f043987e9f37d78b0c8 Mon Sep 17 00:00:00 2001 From: Yrom Date: Thu, 11 Sep 2025 16:55:18 +0800 Subject: [PATCH] fix(infer_v2): Correct the import path of BigVGAN's custom cuda kernel --- indextts/infer_v2.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/indextts/infer_v2.py b/indextts/infer_v2.py index 71c36e1..ab5ef73 100644 --- a/indextts/infer_v2.py +++ b/indextts/infer_v2.py @@ -100,12 +100,12 @@ class IndexTTS2: if self.use_cuda_kernel: # preload the CUDA kernel for BigVGAN try: - from indextts.BigVGAN.alias_free_activation.cuda import load + from indextts.s2mel.modules.bigvgan.alias_free_activation.cuda import activation1d - anti_alias_activation_cuda = load.load() - print(">> Preload custom CUDA kernel for BigVGAN", anti_alias_activation_cuda) - except: + print(">> Preload custom CUDA kernel for BigVGAN", activation1d.anti_alias_activation_cuda) + except Exception as e: print(">> Failed to load custom CUDA kernel for BigVGAN. Falling back to torch.") + print(f"{e!r}") self.use_cuda_kernel = False self.extract_features = SeamlessM4TFeatureExtractor.from_pretrained("facebook/w2v-bert-2.0")