diff --git a/llm/predict/export_model.py b/llm/predict/export_model.py
index 2d7a59ce6904..a7a2a0f347f8 100644
--- a/llm/predict/export_model.py
+++ b/llm/predict/export_model.py
@@ -18,8 +18,8 @@
 
 import paddle
 from paddle.distributed import fleet
-from predict.predictor import ModelArgument, PredictorArgument, create_predictor
 
+from llm.predict.predictor import ModelArgument, PredictorArgument, create_predictor
 from paddlenlp.trainer import PdArgumentParser
 from paddlenlp.trl import llm_utils