PublicMirrors/voice-changer
1
0
Fork 0
mirror of https://github.com/w-okada/voice-changer.git synced 2025-02-02 16:23:58 +03:00
Code Issues Actions Packages Projects Releases Wiki Activity

bugfix: halfprecision

Browse Source
This commit is contained in:
wataru 2023-06-03 22:17:15 +09:00
parent dc7f9a0e15
commit 26a59857b3
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
server/voice_changer/RVC/pipeline/Pipeline.py
View File

@ -165,11 +165,10 @@ class Pipeline(object):
weight /= weight.sum(axis=1, keepdims=True)
npy = np.sum(self.big_npy[ix] * np.expand_dims(weight, axis=2), axis=1)
if self.isHalf is True:
npy = npy.astype("float16")
# recover silient font
npy = np.concatenate([np.zeros([npyOffset, npy.shape[1]]), npy])
if self.isHalf is True:
npy = npy.astype("float16")
feats = (
torch.from_numpy(npy).unsqueeze(0).to(self.device) * index_rate
@ -225,6 +224,7 @@ class Pipeline(object):
).data.to(dtype=torch.int16)
except RuntimeError as e:
if "HALF" in e.__str__().upper():
print("11", e)
raise HalfPrecisionChangingException()
else:
raise e