move model_test back to models/

ModelCloud · Nov 24, 2024 · 69635db · 69635db
1 parent 82db1b0
commit 69635db
Show file tree

Hide file tree

Showing 34 changed files with 34 additions and 32 deletions.
diff --git a/tests/model_test.py → tests/models/model_test.py b/tests/model_test.py → tests/models/model_test.py
diff --git a/tests/models/test_baichuan.py b/tests/models/test_baichuan.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest
+from model_test import ModelTest
 
 
 class TestBaiChuan(ModelTest):

diff --git a/tests/models/test_bloom.py b/tests/models/test_bloom.py
@@ -1,7 +1,7 @@
 
 import torch  # noqa: E402
 
-from tests.model_test import ModelTest
+from model_test import ModelTest
 
 
 class TestBloom(ModelTest):

diff --git a/tests/models/test_chatglm.py b/tests/models/test_chatglm.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest
+from model_test import ModelTest
 
 
 # The official THUDM/chatglm3-6b's tokenization_chatglm.py has compatibility issues with transformers.

diff --git a/tests/models/test_codegen.py b/tests/models/test_codegen.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest
+from model_test import ModelTest
 
 
 class TestCodeGen(ModelTest):

diff --git a/tests/models/test_cohere.py b/tests/models/test_cohere.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest
+from model_test import ModelTest
 
 
 class TestCohere(ModelTest):

diff --git a/tests/models/test_deci.py b/tests/models/test_deci.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest
+from model_test import ModelTest
 
 
 class TestDeci(ModelTest):

diff --git a/tests/models/test_deepseekv2_lite.py b/tests/models/test_deepseekv2_lite.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest
+from model_test import ModelTest
 
 
 class TestDeepseekV2Lite(ModelTest):

diff --git a/tests/models/test_exaone.py b/tests/models/test_exaone.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest
+from model_test import ModelTest
 
 
 class TestExaone(ModelTest):

diff --git a/tests/models/test_falcon.py b/tests/models/test_falcon.py
@@ -1,5 +1,7 @@
 import torch  # noqa: E402from tests.model_test import ModelTest
 
+from model_test import ModelTest
+
 
 class TestFalcon(ModelTest):
     NATIVE_MODEL_ID = "/monster/data/model/falcon-7b-instruct" # "tiiuae/falcon-7b-instruct"

diff --git a/tests/models/test_gemma.py b/tests/models/test_gemma.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest
+from model_test import ModelTest
 
 
 class TestGemma(ModelTest):

diff --git a/tests/models/test_glm.py b/tests/models/test_glm.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest
+from model_test import ModelTest
 
 
 class TestGlm(ModelTest):

diff --git a/tests/models/test_gpt2.py b/tests/models/test_gpt2.py
@@ -1,6 +1,6 @@
 import torch  # noqa: E402
 
-from tests.model_test import ModelTest
+from model_test import ModelTest
 
 
 class TestGpt2(ModelTest):

diff --git a/tests/models/test_gptbigcode.py b/tests/models/test_gptbigcode.py
@@ -1,7 +1,7 @@
 
 import torch  # noqa: E402
 
-from tests.model_test import ModelTest
+from model_test import ModelTest
 
 
 class TestGptBigCode(ModelTest):

diff --git a/tests/models/test_gptj.py b/tests/models/test_gptj.py
@@ -1,7 +1,7 @@
 
 import torch  # noqa: E402
 
-from tests.model_test import ModelTest
+from model_test import ModelTest
 
 
 class TestGptJ(ModelTest):

diff --git a/tests/models/test_gptneox.py b/tests/models/test_gptneox.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest
+from model_test import ModelTest
 
 
 # TODO, this model requires 24G vram at least

diff --git a/tests/models/test_granite.py b/tests/models/test_granite.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest
+from model_test import ModelTest
 
 
 class TestGranite(ModelTest):

diff --git a/tests/models/test_internlm.py b/tests/models/test_internlm.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest
+from model_test import ModelTest
 
 
 class TestInternlm(ModelTest):

diff --git a/tests/models/test_internlm2_5.py b/tests/models/test_internlm2_5.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest
+from model_test import ModelTest
 
 
 class TestInternlm2_5(ModelTest):

diff --git a/tests/models/test_llama3_2.py b/tests/models/test_llama3_2.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest
+from model_test import ModelTest
 
 
 class TestLlama3_2(ModelTest):

diff --git a/tests/models/test_longllama.py b/tests/models/test_longllama.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest
+from model_test import ModelTest
 
 
 class TestLongLlama(ModelTest):

diff --git a/tests/models/test_minicpm.py b/tests/models/test_minicpm.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest
+from model_test import ModelTest
 
 
 class TestMiniCpm(ModelTest):

diff --git a/tests/models/test_mistral.py b/tests/models/test_mistral.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest
+from model_test import ModelTest
 
 
 class TestMistral(ModelTest):

diff --git a/tests/models/test_mixtral.py b/tests/models/test_mixtral.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest
+from model_test import ModelTest
 
 
 class TestMixtral(ModelTest):

diff --git a/tests/models/test_mpt.py b/tests/models/test_mpt.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest
+from model_test import ModelTest
 
 
 class TestMpt(ModelTest):

diff --git a/tests/models/test_opt.py b/tests/models/test_opt.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest
+from model_test import ModelTest
 
 class TestOpt(ModelTest):
     NATIVE_MODEL_ID = "/monster/data/model/opt-125m" # "facebook/opt-125m"

diff --git a/tests/models/test_phi_3.py b/tests/models/test_phi_3.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest
+from model_test import ModelTest
 
 
 class TestPhi_3(ModelTest):

diff --git a/tests/models/test_qwen2_5.py b/tests/models/test_qwen2_5.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest
+from model_test import ModelTest
 
 
 class TestQwen2_5(ModelTest):

diff --git a/tests/models/test_stablelm.py b/tests/models/test_stablelm.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest
+from model_test import ModelTest
 
 
 class TestStablelm(ModelTest):

diff --git a/tests/models/test_starcode2.py b/tests/models/test_starcode2.py
@@ -1,6 +1,6 @@
 import torch  # noqa: E402
 
-from tests.model_test import ModelTest
+from model_test import ModelTest
 
 
 class TestStarCode2(ModelTest):

diff --git a/tests/models/test_tinyllama.py b/tests/models/test_tinyllama.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest
+from model_test import ModelTest
 
 
 class TestTinyllama(ModelTest):

diff --git a/tests/models/test_xverse.py b/tests/models/test_xverse.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest
+from model_test import ModelTest
 
 
 class TestXVerse(ModelTest):

diff --git a/tests/models/test_yi.py b/tests/models/test_yi.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest
+from model_test import ModelTest
 
 
 class TestYi(ModelTest):

diff --git a/tests/test_q4_ipex.py b/tests/test_q4_ipex.py
@@ -4,7 +4,7 @@
 import torch
 
 from gptqmodel import BACKEND
-from tests.model_test import ModelTest
+from .models.model_test import ModelTest
 
 os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
 # -- end do not touch