From 129cfcd9f283dea0d64f2e20b77662febc2d802c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=B8=B8=E9=9B=81?= <zhifu.gzf@alibaba-inc.com>
Date: Thu, 23 Mar 2023 10:01:32 +0800
Subject: [PATCH] cer tool

---
 funasr/export/export_model.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/funasr/export/export_model.py b/funasr/export/export_model.py
index f6ba61616..b1161cbf8 100644
--- a/funasr/export/export_model.py
+++ b/funasr/export/export_model.py
@@ -14,7 +14,7 @@ from funasr.utils.types import str2bool
 # torch_version = float(".".join(torch.__version__.split(".")[:2]))
 # assert torch_version > 1.9
 
-class ASRModelExportParaformer:
+class ModelExport:
     def __init__(
         self,
         cache_dir: Union[Path, str] = None,
@@ -240,7 +240,7 @@ if __name__ == '__main__':
     parser.add_argument('--calib_num', type=int, default=200, help='calib max num')
     args = parser.parse_args()
 
-    export_model = ASRModelExportParaformer(
+    export_model = ModelExport(
         cache_dir=args.export_dir,
         onnx=args.type == 'onnx',
         quant=args.quantize,