Skip to content

Commit 6ee5b27

Browse files
authored
fix test
1 parent b622fe3 commit 6ee5b27

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

test/model/test_model.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,8 @@
11
import os
22
import sys
3+
4+
sys.path.append(os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))))
5+
36
import unittest
47
from model_infer import test_model_inference
58
from lightllm.models.bloom.model import BloomTpPartModel
@@ -22,9 +25,6 @@
2225
from lightllm.utils.config_utils import get_dtype
2326
from lightllm.utils.config_utils import get_config_json
2427

25-
sys.path.append(os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))))
26-
27-
2828
def get_model(weight_dir):
2929
model_cfg = get_config_json(weight_dir)
3030
model_type = model_cfg["model_type"]

0 commit comments

Comments
 (0)