From a801416805d37de1188b7b6dd4e089be3e4b0f7d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=A6=BE=E6=81=AF?= <zhoubofan.zbf@antgroup.com>
Date: Tue, 3 Sep 2024 11:07:47 +0800
Subject: [PATCH] mirror modify

---
 cosyvoice/cli/model.py | 6 +-----
 1 file changed, 1 insertion(+), 5 deletions(-)

diff --git a/cosyvoice/cli/model.py b/cosyvoice/cli/model.py
index 8401d42..c31f281 100644
--- a/cosyvoice/cli/model.py
+++ b/cosyvoice/cli/model.py
@@ -22,11 +22,6 @@ from cosyvoice.utils.common import fade_in_out
 import numpy as np
 import onnxruntime as ort
 
-# try:
-#     import tensorrt as trt
-# except ImportError:
-#     ...
-
 class CosyVoiceModel:
 
     def __init__(self,
@@ -74,6 +69,7 @@ class CosyVoiceModel:
         self.llm.llm = llm_llm
 
     # 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_path = os.path.join(model_dir, trt_file_name)
     #     if not os.path.isfile(trt_file_path):