opensource/ChatTTS
3
0
Fork 0
mirror of https://github.com/2noise/ChatTTS.git synced 2026-09-01 14:55:37 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(npu): ascend sampling real speaker (#788)

Browse Source 
Co-authored-by: shanshan shen <shanshanshen333@gmail.com>
This commit is contained in:
Shanshan Shen
2024-10-21 14:57:07 +08:00
committed by GitHub
parent cff3b4e5c3
commit 0ec82fef45
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ChatTTS/core.py
+4 -2
View File
@@ -268,17 +268,19 @@ class Chat:
self.vocos = vocos
self.logger.log(logging.INFO, "vocos loaded.")
# computation of MelSpectrogram on npu is not support now, use cpu fallback.
dvae_device = torch.device("cpu") if "npu" in str(self.device) else device
dvae = DVAE(
decoder_config=asdict(self.config.dvae.decoder),
encoder_config=asdict(self.config.dvae.encoder),
vq_config=asdict(self.config.dvae.vq),
dim=self.config.dvae.decoder.idim,
coef=coef,
device=device,
device=dvae_device,
)
coef = str(dvae)
assert dvae_ckpt_path, "dvae_ckpt_path should not be None"
dvae.load_pretrained(dvae_ckpt_path, device)
dvae.load_pretrained(dvae_ckpt_path, dvae_device)
self.dvae = dvae.eval()
self.logger.log(logging.INFO, "dvae loaded.")