Skip exllamav2 tests

dottxt-ai · Aug 20, 2024 · 09e4843 · 09e4843
1 parent 8d1fca6
commit 09e4843
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git 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
@@ -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():