diff --git a/tests/model_loader/test_load_mtp.py b/tests/model_loader/test_load_mtp.py index 415cb4699..59438166f 100644 --- a/tests/model_loader/test_load_mtp.py +++ b/tests/model_loader/test_load_mtp.py @@ -16,6 +16,7 @@ import sys import unittest +from pathlib import Path from unittest.mock import Mock import numpy as np @@ -25,7 +26,8 @@ import paddle.distributed.fleet as fleet from fastdeploy.model_executor.layers.embeddings import VocabParallelEmbedding from fastdeploy.model_executor.models.ernie4_5_mtp import Ernie4_5_MTPForCausalLM -sys.path.append("../") +ROOT = Path(__file__).resolve().parents[2] +sys.path.insert(0, str(ROOT)) from tests.utils import get_default_test_fd_config strategy = fleet.DistributedStrategy()