mirror of
https://github.com/FunAudioLLM/CosyVoice.git
synced 2026-02-04 17:39:25 +08:00
8 lines
623 B
Python
8 lines
623 B
Python
# TODO 跟export_jit一样的逻辑，完成flow部分的estimator的onnx导出。
|
||
# tensorrt的安装方式，再这里写一下步骤提示如下，如果没有安装，那么不要执行这个脚本，提示用户先安装，不给选择
|
||
try:
|
||
import tensorrt
|
||
except ImportError:
|
||
print('step1, 下载\n step2. 解压，安装whl，')
|
||
# 安装命令里tensosrt的根目录用环境变量导入，比如os.environ['tensorrt_root_dir']/bin/exetrace，然后python里subprocess里执行导出命令
|
||
# 后面我会在run.sh里写好执行命令 tensorrt_root_dir=xxxx python cosyvoice/bin/export_trt.py --model_dir xxx |