Merge branch '4.1-Fork' of https://github.com/CNChTu/so-vits-svc into 4.1-Fork

This commit is contained in:
ylzz1997 2023-09-04 01:19:09 +08:00
commit b6c779ad2a
1 changed files with 0 additions and 1 deletions

View File

@ -566,7 +566,6 @@ class Volume_Extractor:
audio = torch.Tensor(audio)
n_frames = int(audio.size(-1) // self.hop_size)
audio2 = audio ** 2
audio2 = torch.nn.functional.pad(audio2, (int(self.hop_size // 2), int((self.hop_size + 1) // 2)), mode = 'reflect')
volume = torch.nn.functional.unfold(audio2[:,None,None,:],(1,self.hop_size),stride=self.hop_size)[:,:,:n_frames].mean(dim=1)[0]
volume = torch.sqrt(volume)
return volume