deepgeek/CosyVoice
2
0
Fork 0
mirror of https://github.com/FunAudioLLM/CosyVoice.git synced 2026-02-05 18:09:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix lint

Browse Source
This commit is contained in:
lyuxiang.lx
2025-01-02 12:32:43 +08:00
parent 77d8cf13a3
commit 99ab0f4fcb
2 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
cosyvoice/bin/export_jit.py
View File

@@ -46,6 +46,7 @@ def get_optimized_script(model, preserved_attrs=[]):
script = torch.jit.optimize_for_inference(script) script = torch.jit.optimize_for_inference(script)
return script return script
def main(): def main():
args = get_args() args = get_args()
logging.basicConfig(level=logging.DEBUG, logging.basicConfig(level=logging.DEBUG,
@@ -57,10 +58,10 @@ def main():
try: try:
model = CosyVoice(args.model_dir) model = CosyVoice(args.model_dir)
except: except Exception:
try: try:
model = CosyVoice2(args.model_dir) model = CosyVoice2(args.model_dir)
except: except Exception:
raise TypeError('no valid model_type!') raise TypeError('no valid model_type!')
if not isinstance(model, CosyVoice2): if not isinstance(model, CosyVoice2):

4
cosyvoice/bin/export_onnx.py
View File

@@ -58,10 +58,10 @@ def main():
try: try:
model = CosyVoice(args.model_dir) model = CosyVoice(args.model_dir)
except: except Exception:
try: try:
model = CosyVoice2(args.model_dir) model = CosyVoice2(args.model_dir)
except: except Exception:
raise TypeError('no valid model_type!') raise TypeError('no valid model_type!')
# 1. export flow decoder estimator # 1. export flow decoder estimator