From 09e48437c4d0023e5354e4459639d55888bf02af Mon Sep 17 00:00:00 2001 From: isamu-isozaki Date: Tue, 20 Aug 2024 14:19:23 -0400 Subject: [PATCH] Skip exllamav2 tests --- pyproject.toml | 2 +- tests/generate/conftest.py | 8 +++++--- 2 files changed, 6 insertions(+), 4 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index efcfd5882..52588d929 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -63,7 +63,7 @@ test = [ "torch", "transformers", "pillow", - "exllamav2@git+https://github.com/lapp0/exllamav2.git#egg=sampler-logits-processor", + "exllamav2", ] serve = [ "vllm>=0.3.0", diff --git a/tests/generate/conftest.py b/tests/generate/conftest.py index ed8830119..abd9c72a4 100644 --- a/tests/generate/conftest.py +++ b/tests/generate/conftest.py @@ -27,9 +27,11 @@ def pytest_collection_modifyitems(config, items): for item in items: if "model_fixture" in item.fixturenames: model_param = item.callspec.params.get("model_fixture", None) - if model_param.startswith( - "model_transformers_vision" - ) or model_param.startswith("model_vllm"): + if ( + model_param.startswith("model_transformers_vision") + or model_param.startswith("model_vllm") + or model_param.startswith("model_exllamav2") + ): item.add_marker(skip_marker) if not is_metal_available():