diff --git a/indextts/infer_v2.py b/indextts/infer_v2.py index d59fb81..8e5451b 100644 --- a/indextts/infer_v2.py +++ b/indextts/infer_v2.py @@ -122,7 +122,7 @@ class IndexTTS2: self.semantic_std = self.semantic_std.to(self.device) semantic_codec = build_semantic_codec(self.cfg.semantic_codec) - semantic_code_ckpt = hf_hub_download("amphion/MaskGCT", filename="semantic_codec/model.safetensors") + semantic_code_ckpt = os.path.join(self.model_dir, "semantic_codec/model.safetensors") safetensors.torch.load_model(semantic_codec, semantic_code_ckpt) self.semantic_codec = semantic_codec.to(self.device) self.semantic_codec.eval() @@ -144,16 +144,14 @@ class IndexTTS2: print(">> s2mel weights restored from:", s2mel_path) # load campplus_model - campplus_ckpt_path = hf_hub_download( - "funasr/campplus", filename="campplus_cn_common.bin" - ) + campplus_ckpt_path = os.path.join(self.model_dir, "campplus_cn_common.bin") campplus_model = CAMPPlus(feat_dim=80, embedding_size=192) campplus_model.load_state_dict(torch.load(campplus_ckpt_path, map_location="cpu")) self.campplus_model = campplus_model.to(self.device) self.campplus_model.eval() print(">> campplus_model weights restored from:", campplus_ckpt_path) - bigvgan_name = self.cfg.vocoder.name + bigvgan_name = os.path.join(self.model_dir, self.cfg.vocoder.name) self.bigvgan = bigvgan.BigVGAN.from_pretrained(bigvgan_name, use_cuda_kernel=self.use_cuda_kernel) self.bigvgan = self.bigvgan.to(self.device) self.bigvgan.remove_weight_norm()