half precision to onnx

This commit is contained in:
wataru 2023-04-08 04:03:48 +09:00
parent 60bffb61a5
commit f1bd7db7c1

View File

@ -45,7 +45,7 @@ class ModelWrapper:
audio1 = self.onnx_session.run(
["audio"],
{
"feats": feats.cpu().numpy().astype(np.float32),
"feats": feats.cpu().numpy().astype(np.float16),
"p_len": p_len.cpu().numpy(),
"pitch": pitch.cpu().numpy(),
"pitchf": pitchf.cpu().numpy(),