deepgeek/CosyVoice
2
0
Fork 0
mirror of https://github.com/FunAudioLLM/CosyVoice.git synced 2026-02-04 17:39:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

set use_flow_cache=True when export

Browse Source
This commit is contained in:
lyuxiang.lx
2025-04-07 22:27:17 +08:00
parent 39ffc50dec
commit 7902d1c17f
2 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
cosyvoice/bin/export_jit.py
View File

@@ -61,7 +61,8 @@ def main():
model = CosyVoice(args.model_dir)
except Exception:
try:
model = CosyVoice2(args.model_dir)
# NOTE set use_flow_cache=True when export jit for cache inference
model = CosyVoice2(args.model_dir, use_flow_cache=True)
except Exception:
raise TypeError('no valid model_type!')

3
cosyvoice/bin/export_onnx.py
View File

@@ -62,7 +62,8 @@ def main():
model = CosyVoice(args.model_dir)
except Exception:
try:
model = CosyVoice2(args.model_dir)
# NOTE set use_flow_cache=True when export jit for cache inference
model = CosyVoice2(args.model_dir, use_flow_cache=True)
except Exception:
raise TypeError('no valid model_type!')