From 99ab0f4fcbaf0190eab68b9a4df480e7357fa5dd Mon Sep 17 00:00:00 2001 From: "lyuxiang.lx" Date: Thu, 2 Jan 2025 12:32:43 +0800 Subject: [PATCH] fix lint --- cosyvoice/bin/export_jit.py | 5 +++-- cosyvoice/bin/export_onnx.py | 4 ++-- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/cosyvoice/bin/export_jit.py b/cosyvoice/bin/export_jit.py index 6587b95..ddd486e 100644 --- a/cosyvoice/bin/export_jit.py +++ b/cosyvoice/bin/export_jit.py @@ -46,6 +46,7 @@ def get_optimized_script(model, preserved_attrs=[]): script = torch.jit.optimize_for_inference(script) return script + def main(): args = get_args() logging.basicConfig(level=logging.DEBUG, @@ -57,10 +58,10 @@ def main(): try: model = CosyVoice(args.model_dir) - except: + except Exception: try: model = CosyVoice2(args.model_dir) - except: + except Exception: raise TypeError('no valid model_type!') if not isinstance(model, CosyVoice2): diff --git a/cosyvoice/bin/export_onnx.py b/cosyvoice/bin/export_onnx.py index 52a04b6..9ddd358 100644 --- a/cosyvoice/bin/export_onnx.py +++ b/cosyvoice/bin/export_onnx.py @@ -58,10 +58,10 @@ def main(): try: model = CosyVoice(args.model_dir) - except: + except Exception: try: model = CosyVoice2(args.model_dir) - except: + except Exception: raise TypeError('no valid model_type!') # 1. export flow decoder estimator