Skip to content

Commit 793b8ba

Browse files
committed
fix ernie-csc depoly core dumped
1 parent 2ad9117 commit 793b8ba

File tree

3 files changed

+3
-0
lines changed

3 files changed

+3
-0
lines changed

examples/text_correction/ernie-csc/predict.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -51,6 +51,7 @@ def __init__(self, model_file, params_file, device, max_seq_length, tokenizer, p
5151
# such as enable_mkldnn, set_cpu_math_library_num_threads
5252
config.disable_gpu()
5353
config.switch_use_feed_fetch_ops(False)
54+
config.delete_pass("fused_multi_transformer_encoder_pass")
5455
self.predictor = paddle.inference.create_predictor(config)
5556

5657
self.input_handles = [self.predictor.get_input_handle(name) for name in self.predictor.get_input_names()]

scripts/regression/requirements_ci.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,3 +29,4 @@ cma
2929
paddleocr
3030
fast_tokenizer_python
3131
https://paddle-qa.bj.bcebos.com/PaddleSlim/paddleslim-0.0.0.dev0-py3-none-any.whl
32+
onnxruntime

scripts/regression/run_release.sh

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,7 @@ export log_path=${nlp_dir}/logs
2626
####################################
2727
# for paddlenlp env
2828
python -c 'import sys; print(sys.version_info[:])'
29+
python -c 'import nltk; nltk.download("punkt")'
2930
set -x
3031
python -c "import paddle; print('paddle version:',paddle.__version__,'\npaddle commit:',paddle.version.commit)";
3132
nlp1_build (){

0 commit comments

Comments
 (0)