From 48dceada717e5b471b2dfdb8254794c06c681581 Mon Sep 17 00:00:00 2001 From: ZX-ModelCloud Date: Thu, 19 Dec 2024 07:44:25 +0000 Subject: [PATCH] cleanup --- tests/models/model_test.py | 2 +- tests/models/{ => ovis}/10016.jpg | Bin tests/models/{ => ovis}/ovis_calibration_dataset.py | 0 tests/models/test_ovis_1_6_llama.py | 2 +- 4 files changed, 2 insertions(+), 2 deletions(-) rename tests/models/{ => ovis}/10016.jpg (100%) rename tests/models/{ => ovis}/ovis_calibration_dataset.py (100%) diff --git a/tests/models/model_test.py b/tests/models/model_test.py index 7af9a7bcb..cf70eef3f 100644 --- 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 b/tests/models/ovis/10016.jpg similarity index 100% rename from tests/models/10016.jpg rename to tests/models/ovis/10016.jpg diff --git a/tests/models/ovis_calibration_dataset.py b/tests/models/ovis/ovis_calibration_dataset.py similarity index 100% rename from tests/models/ovis_calibration_dataset.py rename to tests/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 index bd8b72692..699c73efd 100644 --- 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'\n{text}'