Add JAX dependency for tests

dottxt-ai · Nov 28, 2024 · 2ef76ca · 2ef76ca
1 parent 99acf54
commit 2ef76ca
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -70,6 +70,7 @@ test = [
     "transformers",
     "pillow",
     "exllamav2",
+    "jax"
 ]
 serve = [
     "vllm>=0.3.0",