diff --git a/GPT_SoVITS/sv.py b/GPT_SoVITS/sv.py index 00f0cff..fb5806f 100644 --- a/GPT_SoVITS/sv.py +++ b/GPT_SoVITS/sv.py @@ -1,6 +1,6 @@ import sys,os,torch sys.path.append(f"{os.getcwd()}/GPT_SoVITS/eres2net") -sv_path = "GPT_SoVITS\pretrained_models\sv\pretrained_eres2netv2w24s4ep4.ckpt" +sv_path = "GPT_SoVITS/pretrained_models/sv/pretrained_eres2netv2w24s4ep4.ckpt" from ERes2NetV2 import ERes2NetV2 import kaldi as Kaldi class SV: @@ -16,9 +16,9 @@ class SV: self.embedding_model=self.embedding_model.half().to(device) self.is_half=is_half - def compute_embedding3(self,wav):#(1,x)#-1~1 + def compute_embedding3(self,wav): with torch.no_grad(): if self.is_half==True:wav=wav.half() feat = torch.stack([Kaldi.fbank(wav0.unsqueeze(0), num_mel_bins=80, sample_frequency=16000, dither=0) for wav0 in wav]) sv_emb = self.embedding_model.forward3(feat) - return sv_emb \ No newline at end of file + return sv_emb