cleanup

ZX-ModelCloud · Dec 19, 2024 · 48dcead · 48dcead
1 parent a4355b4
commit 48dcead
Show file tree

Hide file tree

Showing 4 changed files with 2 additions and 2 deletions.
diff --git a/tests/models/model_test.py b/tests/models/model_test.py
@@ -22,7 +22,7 @@
 from gptqmodel.quantization import FORMAT  # noqa: E402
 from gptqmodel.quantization.config import QuantizeConfig  # noqa: E402
 from gptqmodel.utils.eval import lm_eval  # noqa: E402
-from ovis_calibration_dataset import get_calib_dataset
+from ovis.ovis_calibration_dataset import get_calib_dataset
 from gptqmodel.utils.torch import torch_empty_cache  # noqa: E402
 
 

diff --git a/tests/models/10016.jpg → tests/models/ovis/10016.jpg b/tests/models/10016.jpg → tests/models/ovis/10016.jpg
diff --git a/tests/models/ovis_calibration_dataset.py → ...s/models/ovis/ovis_calibration_dataset.py b/tests/models/ovis_calibration_dataset.py → ...s/models/ovis/ovis_calibration_dataset.py
diff --git a/tests/models/test_ovis_1_6_llama.py b/tests/models/test_ovis_1_6_llama.py
@@ -28,7 +28,7 @@ def test_ovis_1_6(self):
             visual_tokenizer = model.get_visual_tokenizer()
 
             # enter image path and prompt
-            image_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "10016.jpg")
+            image_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "ovis/10016.jpg")
             image = Image.open(image_path)
             text = "What does this picture show?"
             query = f'<image>\n{text}'