We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent b622fe3 commit 6ee5b27Copy full SHA for 6ee5b27
test/model/test_model.py
@@ -1,5 +1,8 @@
1
import os
2
import sys
3
+
4
+sys.path.append(os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))))
5
6
import unittest
7
from model_infer import test_model_inference
8
from lightllm.models.bloom.model import BloomTpPartModel
@@ -22,9 +25,6 @@
22
25
from lightllm.utils.config_utils import get_dtype
23
26
from lightllm.utils.config_utils import get_config_json
24
27
-sys.path.append(os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))))
-
28
def get_model(weight_dir):
29
model_cfg = get_config_json(weight_dir)
30
model_type = model_cfg["model_type"]
0 commit comments