We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents b4dea3d + 0b75c3a commit 07a3147Copy full SHA for 07a3147
cosyvoice/cli/model.py
@@ -87,6 +87,8 @@ def load_trt(self, flow_decoder_estimator_model, flow_decoder_onnx_model, fp16):
87
assert torch.cuda.is_available(), 'tensorrt only supports gpu!'
88
if not os.path.exists(flow_decoder_estimator_model):
89
convert_onnx_to_trt(flow_decoder_estimator_model, flow_decoder_onnx_model, fp16)
90
+ if os.path.getsize(flow_decoder_estimator_model) == 0:
91
+ raise ValueError('{} is empty file, delete it and export again!'.format(flow_decoder_estimator_model))
92
del self.flow.decoder.estimator
93
import tensorrt as trt
94
with open(flow_decoder_estimator_model, 'rb') as f:
0 commit comments