Spaces:
Running
on
Zero
Running
on
Zero
禾息
commited on
Commit
•
a801416
1
Parent(s):
fadb220
mirror modify
Browse files- cosyvoice/cli/model.py +1 -5
cosyvoice/cli/model.py
CHANGED
|
@@ -22,11 +22,6 @@ from cosyvoice.utils.common import fade_in_out
|
|
| 22 |
import numpy as np
|
| 23 |
import onnxruntime as ort
|
| 24 |
|
| 25 |
-
# try:
|
| 26 |
-
# import tensorrt as trt
|
| 27 |
-
# except ImportError:
|
| 28 |
-
# ...
|
| 29 |
-
|
| 30 |
class CosyVoiceModel:
|
| 31 |
|
| 32 |
def __init__(self,
|
|
@@ -74,6 +69,7 @@ class CosyVoiceModel:
|
|
| 74 |
self.llm.llm = llm_llm
|
| 75 |
|
| 76 |
# def load_trt(self, model_dir, use_fp16):
|
|
|
|
| 77 |
# trt_file_name = 'estimator_fp16.plan' if use_fp16 else 'estimator_fp32.plan'
|
| 78 |
# trt_file_path = os.path.join(model_dir, trt_file_name)
|
| 79 |
# if not os.path.isfile(trt_file_path):
|
|
|
|
| 22 |
import numpy as np
|
| 23 |
import onnxruntime as ort
|
| 24 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 25 |
class CosyVoiceModel:
|
| 26 |
|
| 27 |
def __init__(self,
|
|
|
|
| 69 |
self.llm.llm = llm_llm
|
| 70 |
|
| 71 |
# def load_trt(self, model_dir, use_fp16):
|
| 72 |
+
# import tensorrt as trt
|
| 73 |
# trt_file_name = 'estimator_fp16.plan' if use_fp16 else 'estimator_fp32.plan'
|
| 74 |
# trt_file_path = os.path.join(model_dir, trt_file_name)
|
| 75 |
# if not os.path.isfile(trt_file_path):
|