mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-10-24 09:03:43 +08:00
fix import tests.utils error in tests/model_loader/test_load_mtp.py (#4027)
Co-authored-by: yongqiangma <xing.wo@163.com>
This commit is contained in:
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user