tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Update infer.py

Browse Source
This commit is contained in:
Stardust·减
2023-11-25 13:49:17 +08:00
committed by GitHub
parent 3ec1b6559f
commit 64e5746ff7
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
infer.py
View File

@@ -216,6 +216,7 @@ def infer(
ja_bert = ja_bert.to(device).unsqueeze(0) ja_bert = ja_bert.to(device).unsqueeze(0)
en_bert = en_bert.to(device).unsqueeze(0) en_bert = en_bert.to(device).unsqueeze(0)
x_tst_lengths = torch.LongTensor([phones.size(0)]).to(device) x_tst_lengths = torch.LongTensor([phones.size(0)]).to(device)
emo = emo.to(device).unsqueeze(0)
del phones del phones
speakers = torch.LongTensor([hps.data.spk2id[sid]]).to(device) speakers = torch.LongTensor([hps.data.spk2id[sid]]).to(device)
audio = ( audio = (
@@ -228,6 +229,7 @@ def infer(
bert, bert,
ja_bert, ja_bert,
en_bert, en_bert,
emo,
sdp_ratio=sdp_ratio, sdp_ratio=sdp_ratio,
noise_scale=noise_scale, noise_scale=noise_scale,
noise_scale_w=noise_scale_w, noise_scale_w=noise_scale_w,
@@ -237,7 +239,7 @@ def infer(
.float() .float()
.numpy() .numpy()
) )
del x_tst, tones, lang_ids, bert, x_tst_lengths, speakers, ja_bert, en_bert del x_tst, tones, lang_ids, bert, x_tst_lengths, speakers, ja_bert, en_bert, emo
if torch.cuda.is_available(): if torch.cuda.is_available():
torch.cuda.empty_cache() torch.cuda.empty_cache()
return audio return audio