From f60266d6330dcdd2a40f16938bde404dc00e7909 Mon Sep 17 00:00:00 2001 From: nyx-c-language Date: Wed, 23 Apr 2025 10:32:54 +0800 Subject: [PATCH] fix pir --- slm/examples/lexical_analysis/deploy/predict.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/slm/examples/lexical_analysis/deploy/predict.py b/slm/examples/lexical_analysis/deploy/predict.py index 79fda25f95f7..0253d1802c41 100644 --- a/slm/examples/lexical_analysis/deploy/predict.py +++ b/slm/examples/lexical_analysis/deploy/predict.py @@ -19,11 +19,15 @@ import paddle from paddlenlp.data import Pad, Stack, Tuple +from paddlenlp.utils.env import ( + PADDLE_INFERENCE_MODEL_SUFFIX, + PADDLE_INFERENCE_WEIGHTS_SUFFIX, +) # yapf: disable parser = argparse.ArgumentParser(__doc__) -parser.add_argument("--model_file", type=str, required=True, default='./static_graph_params.pdmodel', help="The path to model info in static graph.") -parser.add_argument("--params_file", type=str, required=True, default='./static_graph_params.pdiparams', help="The path to parameters in static graph.") +parser.add_argument("--model_file", type=str, required=True, default=f'./static_graph_params{PADDLE_INFERENCE_MODEL_SUFFIX}', help="The path to model info in static graph.") +parser.add_argument("--params_file", type=str, required=True, default=f'./static_graph_params{PADDLE_INFERENCE_WEIGHTS_SUFFIX}', help="The path to parameters in static graph.") parser.add_argument("--data_dir", type=str, default=None, help="The folder where the dataset is located.") parser.add_argument("--init_checkpoint", type=str, default=None, help="Path to init model.") parser.add_argument("--batch_size", type=int, default=2, help="The number of sequences contained in a mini-batch.")