Katock commited on
Commit
f8edc9e
·
1 Parent(s): 527e4e0

Update infer_tool.py

Browse files
Files changed (1) hide show
  1. inference/infer_tool.py +1 -1
inference/infer_tool.py CHANGED
@@ -179,7 +179,7 @@ class Svc(object):
179
 
180
  wav16k = librosa.resample(wav, orig_sr=self.target_sample, target_sr=16000)
181
  wav16k = torch.from_numpy(wav16k).to(self.dev)
182
- c = self.hubert_model.encoder(wav16k)
183
  c = utils.repeat_expand_2d(c.squeeze(0), f0.shape[1])
184
 
185
  if cluster_infer_ratio !=0:
 
179
 
180
  wav16k = librosa.resample(wav, orig_sr=self.target_sample, target_sr=16000)
181
  wav16k = torch.from_numpy(wav16k).to(self.dev)
182
+ c = utils.get_hubert_content(self.hubert_model, wav_16k_tensor=wav16k)
183
  c = utils.repeat_expand_2d(c.squeeze(0), f0.shape[1])
184
 
185
  if cluster_infer_ratio !=0: