mirror of
https://github.com/w-okada/voice-changer.git
synced 2025-02-03 00:33:57 +03:00
half precision to onnx
This commit is contained in:
parent
60bffb61a5
commit
f1bd7db7c1
@ -45,7 +45,7 @@ class ModelWrapper:
|
|||||||
audio1 = self.onnx_session.run(
|
audio1 = self.onnx_session.run(
|
||||||
["audio"],
|
["audio"],
|
||||||
{
|
{
|
||||||
"feats": feats.cpu().numpy().astype(np.float32),
|
"feats": feats.cpu().numpy().astype(np.float16),
|
||||||
"p_len": p_len.cpu().numpy(),
|
"p_len": p_len.cpu().numpy(),
|
||||||
"pitch": pitch.cpu().numpy(),
|
"pitch": pitch.cpu().numpy(),
|
||||||
"pitchf": pitchf.cpu().numpy(),
|
"pitchf": pitchf.cpu().numpy(),
|
||||||
|
Loading…
Reference in New Issue
Block a user