Update ci_case.sh

PaddlePaddle · Feb 9, 2023 · b258a8e · b258a8e
1 parent 82856d8
commit b258a8e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/model_zoo/ernie-m/deploy/predictor/inference.py b/model_zoo/ernie-m/deploy/predictor/inference.py
@@ -14,10 +14,9 @@
 
 import argparse
 
+from ernie_m_predictor import ErnieMPredictor
 from psutil import cpu_count
 
-from .ernie_m_predictor import ErnieMPredictor
-
 
 def parse_args():
     parser = argparse.ArgumentParser()

diff --git a/scripts/regression/ci_case.sh b/scripts/regression/ci_case.sh
@@ -994,7 +994,7 @@ if [ ! -f 'test.py' ];then
         --remove_unused_columns False >${log_path}/ernie-m_clt >>${log_path}/ernie-m_clt 2>&1
     print_info $? ernie-m_clt
     # inference for cross-lingual-transfer
-    python deploy/predictor/inference.py 
+    python deploy/predictor/inference.py \
         --device cpu \
         --task_name seq_cls \
         --model_path output_clt/export/model >${log_path}/ernie-m_clt_infer >>${log_path}/ernie-m_clt_infer 2>&1
@@ -1016,7 +1016,7 @@ if [ ! -f 'test.py' ];then
         --remove_unused_columns False >${log_path}/ernie-m_tta >>${log_path}/ernie-m_tta 2>&1
     print_info $? ernie-m_tta
     # inference for translate-train-all
-    python deploy/predictor/inference.py 
+    python deploy/predictor/inference.py \
         --device cpu \
         --task_name seq_cls \
         --model_path output_tta/export/model >${log_path}/ernie-m_tta_infer >>${log_path}/ernie-m_tta_infer 2>&1