From a7d7a0f3a2e7cd44a337ced34e3536b12ccb534e Mon Sep 17 00:00:00 2001
From: zhifu gao <zhifu.gzf@alibaba-inc.com>
Date: 星期一, 11 三月 2024 19:24:44 +0800
Subject: [PATCH] Dev gzf (#1467)

---
 examples/industrial_data_pretraining/ct_transformer/export.sh |    6 ++++--
 1 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/examples/industrial_data_pretraining/ct_transformer/export.sh b/examples/industrial_data_pretraining/ct_transformer/export.sh
index 7556458..a11cda5 100644
--- a/examples/industrial_data_pretraining/ct_transformer/export.sh
+++ b/examples/industrial_data_pretraining/ct_transformer/export.sh
@@ -12,7 +12,8 @@
 ++model=${model} \
 ++model_revision=${model_revision} \
 ++type="onnx" \
-++quantize=false
+++quantize=false \
+++device="cpu"
 
 
 # method2, inference from local path
@@ -21,4 +22,5 @@
 python -m funasr.bin.export \
 ++model=${model} \
 ++type="onnx" \
-++quantize=false
\ No newline at end of file
+++quantize=false \
+++device="cpu"
\ No newline at end of file

--
Gitblit v1.9.1