From e0a6daebf2646b0612b13e12979a2bb7164c54b8 Mon Sep 17 00:00:00 2001
From: root <zhangyuekai@foxmail.com>
Date: 星期一, 27 二月 2023 17:57:53 +0800
Subject: [PATCH] add triton
---
funasr/export/README.md | 15 ++++-----------
1 files changed, 4 insertions(+), 11 deletions(-)
diff --git a/funasr/export/README.md b/funasr/export/README.md
index 9740f23..0ecf272 100644
--- a/funasr/export/README.md
+++ b/funasr/export/README.md
@@ -16,17 +16,13 @@
output_dir = "../export" # onnx/torchscripts model save path
export_model = ASRModelExportParaformer(cache_dir=output_dir, onnx=True)
-export_model.export_from_modelscope('damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch')
+export_model.export('damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch')
```
Export model from local path
```python
-from funasr.export.export_model import ASRModelExportParaformer
-
-output_dir = "../export" # onnx/torchscripts model save path
-export_model = ASRModelExportParaformer(cache_dir=output_dir, onnx=True)
-export_model.export_from_local('/root/cache/export/damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch')
+export_model.export('/root/cache/export/damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch')
```
## Export torchscripts format model
@@ -36,15 +32,12 @@
output_dir = "../export" # onnx/torchscripts model save path
export_model = ASRModelExportParaformer(cache_dir=output_dir, onnx=False)
-export_model.export_from_modelscope('damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch')
+export_model.export('damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch')
```
Export model from local path
```python
-from funasr.export.export_model import ASRModelExportParaformer
-output_dir = "../export" # onnx/torchscripts model save path
-export_model = ASRModelExportParaformer(cache_dir=output_dir, onnx=False)
-export_model.export_from_local('/root/cache/export/damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch')
+export_model.export('/root/cache/export/damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch')
```
--
Gitblit v1.9.1