1berry/wokada-voice-changer
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

mps

Browse Source
This commit is contained in:
wataru 2023-04-18 04:32:43 +09:00
parent a7e907370d
commit 7d96ab4264
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
server/voice_changer/RVC/RVC.py
View File

@ -250,7 +250,7 @@ class RVC:
else:
dev = torch.device("cuda", index=self.settings.gpu)
print("device:", dev)
# print("device:", dev)
self.hubert_model = self.hubert_model.to(dev)
self.net_g = self.net_g.to(dev)

3
server/voice_changer/RVC/custom_vc_infer_pipeline.py
View File

@ -142,7 +142,8 @@ class VC(object):
pitch = pitch[:p_len]
pitchf = pitchf[:p_len]
pitch = torch.tensor(pitch, device=self.device).unsqueeze(0).long()
pitchf = torch.tensor(pitchf, device=self.device).unsqueeze(0).float()
pitchf = torch.tensor(pitchf, device=self.device, dtype=torch.float).unsqueeze(0)
t2 = ttime()
times[1] += (t2 - t1)
if self.t_pad_tgt == 0: