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

mirror modify

Browse Source
This commit is contained in:
禾息
2024-09-03 11:07:47 +08:00
parent fadb22086f
commit a801416805
1 changed files with 1 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
cosyvoice/cli/model.py
View File

@@ -22,11 +22,6 @@ from cosyvoice.utils.common import fade_in_out
import numpy as np import numpy as np
import onnxruntime as ort import onnxruntime as ort
# try:
# import tensorrt as trt
# except ImportError:
# ...
class CosyVoiceModel: class CosyVoiceModel:
def __init__(self, def __init__(self,
@@ -74,6 +69,7 @@ class CosyVoiceModel:
self.llm.llm = llm_llm self.llm.llm = llm_llm
# def load_trt(self, model_dir, use_fp16): # def load_trt(self, model_dir, use_fp16):
# import tensorrt as trt
# trt_file_name = 'estimator_fp16.plan' if use_fp16 else 'estimator_fp32.plan' # trt_file_name = 'estimator_fp16.plan' if use_fp16 else 'estimator_fp32.plan'
# trt_file_path = os.path.join(model_dir, trt_file_name) # trt_file_path = os.path.join(model_dir, trt_file_name)
# if not os.path.isfile(trt_file_path): # if not os.path.isfile(trt_file_path):