snicolast/ComfyUI-IndexTTS2
1
0
Fork 0
mirror of https://github.com/snicolast/ComfyUI-IndexTTS2.git synced 2026-01-26 14:39:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update infer_v2.py

Browse Source 
indextts/infer_v2.py
    - Line 156: Changed bigvgan_name = self.cfg.vocoder.name to bigvgan_name = os.path.join(self.model_dir,
This commit is contained in:
drbaph
2025-09-13 20:59:39 +01:00
committed by GitHub
parent 9dae0730b9
commit 5c35293441
1 changed files with 3 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
indextts/infer_v2.py
View File

@@ -122,7 +122,7 @@ class IndexTTS2:
self.semantic_std = self.semantic_std.to(self.device) self.semantic_std = self.semantic_std.to(self.device)
semantic_codec = build_semantic_codec(self.cfg.semantic_codec) 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) safetensors.torch.load_model(semantic_codec, semantic_code_ckpt)
self.semantic_codec = semantic_codec.to(self.device) self.semantic_codec = semantic_codec.to(self.device)
self.semantic_codec.eval() self.semantic_codec.eval()
@@ -144,16 +144,14 @@ class IndexTTS2:
print(">> s2mel weights restored from:", s2mel_path) print(">> s2mel weights restored from:", s2mel_path)
# load campplus_model # load campplus_model
campplus_ckpt_path = hf_hub_download( campplus_ckpt_path = os.path.join(self.model_dir, "campplus_cn_common.bin")
"funasr/campplus", filename="campplus_cn_common.bin"
)
campplus_model = CAMPPlus(feat_dim=80, embedding_size=192) campplus_model = CAMPPlus(feat_dim=80, embedding_size=192)
campplus_model.load_state_dict(torch.load(campplus_ckpt_path, map_location="cpu")) campplus_model.load_state_dict(torch.load(campplus_ckpt_path, map_location="cpu"))
self.campplus_model = campplus_model.to(self.device) self.campplus_model = campplus_model.to(self.device)
self.campplus_model.eval() self.campplus_model.eval()
print(">> campplus_model weights restored from:", campplus_ckpt_path) 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 = bigvgan.BigVGAN.from_pretrained(bigvgan_name, use_cuda_kernel=self.use_cuda_kernel)
self.bigvgan = self.bigvgan.to(self.device) self.bigvgan = self.bigvgan.to(self.device)
self.bigvgan.remove_weight_norm() self.bigvgan.remove_weight_norm()