From 7d36bb15eea309154a348557b92cd5ab04c9c477 Mon Sep 17 00:00:00 2001 From: intellinjun Date: Thu, 18 Jan 2024 16:02:47 +0800 Subject: [PATCH] fix whisper ci error Signed-off-by: intellinjun --- tests/model-test/cpp_graph_inference.sh | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/model-test/cpp_graph_inference.sh b/tests/model-test/cpp_graph_inference.sh index de910c838..621d71921 100644 --- a/tests/model-test/cpp_graph_inference.sh +++ b/tests/model-test/cpp_graph_inference.sh @@ -390,8 +390,8 @@ function main() { export LANG=en_US.UTF-8 export LC_ALL=en_US.UTF-8 echo "======= Inference Start =======" - if [[ "${model}" == "whisper" ]];then OMP_NUM_THREADS=$cores_per_instance numactl -m 0 -C 0-$(($cores_per_instance - 1)) \ - NEURAL_SPEED_VERBOSE=1 $infer_cmd -f "/tf_dataset2/models/nlp_toolkit/whisper-tiny/jfk.wav" -m ${model}-${precision}.bin + if [[ "${model}" == "whisper" ]];then NEURAL_SPEED_VERBOSE=1 OMP_NUM_THREADS=$cores_per_instance numactl -m 0 -C 0-$(($cores_per_instance - 1)) \ + $infer_cmd -f "/tf_dataset2/models/nlp_toolkit/whisper-tiny/jfk.wav" -m ${model}-${precision}.bin else real_ctx=$ctx # TODO(Zhenzhong): use same ctx for chatglm & baichuan [[ "${model}" == "chatglm2" || "${model}" == "chatglm-6b" ||