This commit is contained in:
ylzz1997 2023-05-29 02:11:04 +08:00
parent 4e982eb7d5
commit 5f25534081
1 changed files with 1 additions and 1 deletions

View File

@ -239,7 +239,7 @@ class Svc(object):
start = time.time()
vol = None
if not self.only_diffusion:
vol = self.volume_extractor.extract(audio[None,:])[None,:].to(self.dev) if self.vol_embedding else None
vol = self.volume_extractor.extract(torch.FloatTensor(wav).to(self.dev)[None,:])[None,:].to(self.dev) if self.vol_embedding else None
audio,f0 = self.net_g_ms.infer(c, f0=f0, g=sid, uv=uv, predict_f0=auto_predict_f0, noice_scale=noice_scale,vol=vol)
audio = audio[0,0].data.float()
audio_mel = self.vocoder.extract(audio[None,:],self.target_sample) if self.shallow_diffusion else None