Fix predict import. (#10476)

Author: ZHUI

llm/predict/export_model.py

@@ -14,12 +14,16 @@
 from __future__ import annotations
 
 import os
+import sys
 from dataclasses import dataclass, field
+from pathlib import Path
 
 import paddle
 from paddle.distributed import fleet
 
-from llm.predict.predictor import ModelArgument, PredictorArgument, create_predictor
+sys.path.append(str(Path(__file__).parent.parent))
+from predict.predictor import ModelArgument, PredictorArgument, create_predictor
+
 from paddlenlp.trainer import PdArgumentParser
 from paddlenlp.trl import llm_utils
 

llm/predict/flask_server.py

@@ -17,11 +17,16 @@
 import json
 import os
 import socket
+import sys
 import time
 from contextlib import closing
 from dataclasses import asdict, dataclass, field
+from pathlib import Path
 from time import sleep
 
+sys.path.append(str(Path(__file__).parent))
+sys.path.append(str(Path(__file__).parent.parent))
+
 import requests
 from filelock import FileLock
 from predict.predictor import (