1
0
lyuxiang.lx 11 сар өмнө
parent
commit
99ab0f4fcb

+ 3 - 2
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):

+ 2 - 2
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