From f2ed4b3856eaed8abe568e6904ffd8dc3a799f5f Mon Sep 17 00:00:00 2001
From: 彭震东 <zhendong.peng@qq.com>
Date: 星期一, 15 七月 2024 17:54:27 +0800
Subject: [PATCH] fix progress bar for batch_size (#1917)
---
examples/industrial_data_pretraining/paraformer/export.py | 2 +-
1 files changed, 1 insertions(+), 1 deletions(-)
diff --git a/examples/industrial_data_pretraining/paraformer/export.py b/examples/industrial_data_pretraining/paraformer/export.py
index a91e9e4..6334e3b 100644
--- a/examples/industrial_data_pretraining/paraformer/export.py
+++ b/examples/industrial_data_pretraining/paraformer/export.py
@@ -13,7 +13,7 @@
model="iic/speech_paraformer-large-contextual_asr_nat-zh-cn-16k-common-vocab8404",
)
-res = model.export(type="torchscripts", quantize=False)
+res = model.export(type="torchscript", quantize=False)
# res = model.export(type="bladedisc", input=f"{model.model_path}/example/asr_example.wav")
print(res)
--
Gitblit v1.9.1