Skip to content

Commit

Permalink
remove ..
Browse files Browse the repository at this point in the history
  • Loading branch information
CSY-ModelCloud committed Nov 24, 2024
1 parent c23b7af commit 2e1e184
Show file tree
Hide file tree
Showing 31 changed files with 31 additions and 31 deletions.
2 changes: 1 addition & 1 deletion tests/models/test_baichuan.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
sys.path.insert(0, parent_dir)

from ..model_test import ModelTest # noqa: E402
from model_test import ModelTest


class TestBaiChuan(ModelTest):
Expand Down
2 changes: 1 addition & 1 deletion tests/models/test_bloom.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@
parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
sys.path.insert(0, parent_dir)

from ..model_test import ModelTest # noqa: E402
from model_test import ModelTest


class TestBloom(ModelTest):
Expand Down
2 changes: 1 addition & 1 deletion tests/models/test_chatglm.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
sys.path.insert(0, parent_dir)

from ..model_test import ModelTest # noqa: E402
from model_test import ModelTest


# The official THUDM/chatglm3-6b's tokenization_chatglm.py has compatibility issues with transformers.
Expand Down
2 changes: 1 addition & 1 deletion tests/models/test_codegen.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
sys.path.insert(0, parent_dir)

from ..model_test import ModelTest # noqa: E402
from model_test import ModelTest


class TestCodeGen(ModelTest):
Expand Down
2 changes: 1 addition & 1 deletion tests/models/test_cohere.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
sys.path.insert(0, parent_dir)

from ..model_test import ModelTest # noqa: E402
from model_test import ModelTest


class TestCohere(ModelTest):
Expand Down
2 changes: 1 addition & 1 deletion tests/models/test_deci.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
sys.path.insert(0, parent_dir)

from ..model_test import ModelTest # noqa: E402
from model_test import ModelTest


class TestDeci(ModelTest):
Expand Down
2 changes: 1 addition & 1 deletion tests/models/test_exaone.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
sys.path.insert(0, parent_dir)

from ..model_test import ModelTest # noqa: E402
from model_test import ModelTest


class TestExaone(ModelTest):
Expand Down
2 changes: 1 addition & 1 deletion tests/models/test_falcon.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@
parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
sys.path.insert(0, parent_dir)

from ..model_test import ModelTest # noqa: E402
from model_test import ModelTest


class TestFalcon(ModelTest):
Expand Down
2 changes: 1 addition & 1 deletion tests/models/test_gemma.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
sys.path.insert(0, parent_dir)

from ..model_test import ModelTest # noqa: E402
from model_test import ModelTest


class TestGemma(ModelTest):
Expand Down
2 changes: 1 addition & 1 deletion tests/models/test_glm.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
sys.path.insert(0, parent_dir)

from ..model_test import ModelTest # noqa: E402
from model_test import ModelTest


class TestGlm(ModelTest):
Expand Down
2 changes: 1 addition & 1 deletion tests/models/test_gpt2.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@
parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
sys.path.insert(0, parent_dir)

from ..model_test import ModelTest # noqa: E402
from model_test import ModelTest


class TestGpt2(ModelTest):
Expand Down
2 changes: 1 addition & 1 deletion tests/models/test_gptbigcode.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@
parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
sys.path.insert(0, parent_dir)

from ..model_test import ModelTest # noqa: E402
from model_test import ModelTest


class TestGptBigCode(ModelTest):
Expand Down
2 changes: 1 addition & 1 deletion tests/models/test_gptj.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@
parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
sys.path.insert(0, parent_dir)

from ..model_test import ModelTest # noqa: E402
from model_test import ModelTest


class TestGptJ(ModelTest):
Expand Down
2 changes: 1 addition & 1 deletion tests/models/test_gptneox.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
sys.path.insert(0, parent_dir)

from ..model_test import ModelTest # noqa: E402
from model_test import ModelTest


# TODO, this model requires 24G vram at least
Expand Down
2 changes: 1 addition & 1 deletion tests/models/test_granite.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
sys.path.insert(0, parent_dir)

from ..model_test import ModelTest # noqa: E402
from model_test import ModelTest


class TestGranite(ModelTest):
Expand Down
2 changes: 1 addition & 1 deletion tests/models/test_internlm.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
sys.path.insert(0, parent_dir)

from ..model_test import ModelTest # noqa: E402
from model_test import ModelTest


class TestInternlm(ModelTest):
Expand Down
2 changes: 1 addition & 1 deletion tests/models/test_internlm2_5.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
sys.path.insert(0, parent_dir)

from ..model_test import ModelTest # noqa: E402
from model_test import ModelTest


class TestInternlm2_5(ModelTest):
Expand Down
2 changes: 1 addition & 1 deletion tests/models/test_llama3_2.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
sys.path.insert(0, parent_dir)

from ..model_test import ModelTest # noqa: E402
from model_test import ModelTest


class TestLlama3_2(ModelTest):
Expand Down
2 changes: 1 addition & 1 deletion tests/models/test_longllama.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
sys.path.insert(0, parent_dir)

from ..model_test import ModelTest # noqa: E402
from model_test import ModelTest


class TestLongLlama(ModelTest):
Expand Down
2 changes: 1 addition & 1 deletion tests/models/test_minicpm.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
sys.path.insert(0, parent_dir)

from ..model_test import ModelTest # noqa: E402
from model_test import ModelTest


class TestMiniCpm(ModelTest):
Expand Down
2 changes: 1 addition & 1 deletion tests/models/test_mistral.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
sys.path.insert(0, parent_dir)

from ..model_test import ModelTest # noqa: E402
from model_test import ModelTest


class TestMistral(ModelTest):
Expand Down
2 changes: 1 addition & 1 deletion tests/models/test_mixtral.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
sys.path.insert(0, parent_dir)

from ..model_test import ModelTest # noqa: E402
from model_test import ModelTest


class TestMixtral(ModelTest):
Expand Down
2 changes: 1 addition & 1 deletion tests/models/test_mpt.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
sys.path.insert(0, parent_dir)

from ..model_test import ModelTest # noqa: E402
from model_test import ModelTest


class TestMpt(ModelTest):
Expand Down
2 changes: 1 addition & 1 deletion tests/models/test_opt.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
sys.path.insert(0, parent_dir)

from ..model_test import ModelTest # noqa: E402
from model_test import ModelTest


class TestOpt(ModelTest):
Expand Down
2 changes: 1 addition & 1 deletion tests/models/test_phi_3.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
sys.path.insert(0, parent_dir)

from ..model_test import ModelTest # noqa: E402
from model_test import ModelTest


class TestPhi_3(ModelTest):
Expand Down
2 changes: 1 addition & 1 deletion tests/models/test_qwen2_5.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
sys.path.insert(0, parent_dir)

from ..model_test import ModelTest # noqa: E402
from model_test import ModelTest


class TestQwen2_5(ModelTest):
Expand Down
2 changes: 1 addition & 1 deletion tests/models/test_stablelm.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
sys.path.insert(0, parent_dir)

from ..model_test import ModelTest # noqa: E402
from model_test import ModelTest


class TestStablelm(ModelTest):
Expand Down
2 changes: 1 addition & 1 deletion tests/models/test_starcode2.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@
parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
sys.path.insert(0, parent_dir)

from ..model_test import ModelTest # noqa: E402
from model_test import ModelTest


class TestStarCode2(ModelTest):
Expand Down
2 changes: 1 addition & 1 deletion tests/models/test_tinyllama.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
sys.path.insert(0, parent_dir)

from ..model_test import ModelTest # noqa: E402
from model_test import ModelTest


class TestTinyllama(ModelTest):
Expand Down
2 changes: 1 addition & 1 deletion tests/models/test_xverse.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
sys.path.insert(0, parent_dir)

from ..model_test import ModelTest # noqa: E402
from model_test import ModelTest


class TestXVerse(ModelTest):
Expand Down
2 changes: 1 addition & 1 deletion tests/models/test_yi.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
sys.path.insert(0, parent_dir)

from ..model_test import ModelTest # noqa: E402
from model_test import ModelTest


class TestYi(ModelTest):
Expand Down

0 comments on commit 2e1e184

Please sign in to comment.