1
0
lyuxiang.lx vor 8 Monaten
Ursprung
Commit
0b75c3a03f
1 geänderte Dateien mit 2 neuen und 0 gelöschten Zeilen
  1. 2 0
      cosyvoice/cli/model.py

+ 2 - 0
cosyvoice/cli/model.py

@@ -87,6 +87,8 @@ class CosyVoiceModel:
         assert torch.cuda.is_available(), 'tensorrt only supports gpu!'
         if not os.path.exists(flow_decoder_estimator_model):
             convert_onnx_to_trt(flow_decoder_estimator_model, flow_decoder_onnx_model, fp16)
+        if os.path.getsize(flow_decoder_estimator_model) == 0:
+            raise ValueError('{} is empty file, delete it and export again!'.format(flow_decoder_estimator_model))
         del self.flow.decoder.estimator
         import tensorrt as trt
         with open(flow_decoder_estimator_model, 'rb') as f: