lyuxiang.lx 11 달 전
부모
커밋
99ab0f4fcb
2개의 변경된 파일5개의 추가작업 그리고 4개의 파일을 삭제
  1. 3 2
      cosyvoice/bin/export_jit.py
  2. 2 2
      cosyvoice/bin/export_onnx.py

+ 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