From 8406f4419561b57dea02a2b3dd4c5d6f4cbb62dd Mon Sep 17 00:00:00 2001 From: Andrew Lapp Date: Tue, 25 Jun 2024 02:44:38 -0500 Subject: [PATCH] Update , enable requirement only on darwin, disable requirement on darwin --- pyproject.toml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index 779add59e..aa88fcbbc 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -56,10 +56,10 @@ test = [ "beartype<0.16.0", "responses", "llama-cpp-python", - "mlx-lm", + "mlx-lm; platform_machine == 'arm64' and sys_platform == 'darwin'", "huggingface_hub", "openai>=1.0.0", - "vllm", + "vllm; sys_platform != 'darwin'", "torch", "transformers", ]