diff --git a/.github/actions/nm-build-vllm/action.yml b/.github/actions/nm-build-vllm/action.yml index baea7498b943d..f068c7a62378d 100644 --- a/.github/actions/nm-build-vllm/action.yml +++ b/.github/actions/nm-build-vllm/action.yml @@ -25,7 +25,7 @@ runs: COMMIT=${{ github.sha }} VENV="${{ inputs.venv }}-${COMMIT:0:7}" source $(pyenv root)/versions/${{ inputs.python }}/envs/${VENV}/bin/activate - pip3 install --index-url http://${{ inputs.pypi }}:8080/ --trusted-host ${{ inputs.pypi }} magic-wand + pip3 install --index-url http://${{ inputs.pypi }}:8080/ --trusted-host ${{ inputs.pypi }} nm-magic-wand pip3 install -r requirements.txt SUCCESS=0 pip3 install -e . || SUCCESS=$? diff --git a/.github/actions/nm-test-vllm/action.yml b/.github/actions/nm-test-vllm/action.yml index 6f7854d9f804b..cb2445f398d4e 100644 --- a/.github/actions/nm-test-vllm/action.yml +++ b/.github/actions/nm-test-vllm/action.yml @@ -28,7 +28,7 @@ runs: COMMIT=${{ github.sha }} VENV="${{ inputs.venv }}-${COMMIT:0:7}" source $(pyenv root)/versions/${{ inputs.python }}/envs/${VENV}/bin/activate - pip3 install --index-url http://${{ inputs.pypi }}:8080/ --trusted-host ${{ inputs.pypi }} magic-wand + pip3 install --index-url http://${{ inputs.pypi }}:8080/ --trusted-host ${{ inputs.pypi }} nm-magic-wand pip3 install -r requirements-dev.txt # run tests via runner script SUCCESS=0 diff --git a/.github/scripts/build b/.github/scripts/build index 62be45bceb36a..9a26c86b35d8e 100755 --- a/.github/scripts/build +++ b/.github/scripts/build @@ -55,6 +55,6 @@ if [ -z "${VENV}" ]; then fi source $(pyenv root)/versions/${PYTHON}/envs/${VENV}/bin/activate -pip3 install --index-url http://${PYPI_IP}:8080/ --trusted-host ${PYPI_IP} magic-wand +pip3 install --index-url http://${PYPI_IP}:8080/ --trusted-host ${PYPI_IP} nm-magic-wand pip3 install -r requirements.txt pip3 install -e . diff --git a/setup.py b/setup.py index c212be75e594f..4d45591064d07 100644 --- a/setup.py +++ b/setup.py @@ -436,7 +436,7 @@ def get_requirements() -> List[str]: return requirements -_sparsity_deps = ["magic_wand"] +_sparsity_deps = ["nm-magic-wand"] def get_extra_requirements() -> dict: diff --git a/vllm/model_executor/layers/parameters/lazy_compressed.py b/vllm/model_executor/layers/parameters/lazy_compressed.py index a9598affcc593..d5ee86ee8fb56 100644 --- a/vllm/model_executor/layers/parameters/lazy_compressed.py +++ b/vllm/model_executor/layers/parameters/lazy_compressed.py @@ -29,7 +29,7 @@ def __new__(cls, if not is_magic_wand_available: raise ValueError( "magic_wand is not available and required for sparsity " - "support. Please install it with `pip install magic_wand`") + "support. Please install it with `pip install nm-magic-wand`") self = torch.Tensor._make_wrapper_subclass( cls, diff --git a/vllm/model_executor/layers/sparsity/__init__.py b/vllm/model_executor/layers/sparsity/__init__.py index ee430e27a0186..874819f343373 100644 --- a/vllm/model_executor/layers/sparsity/__init__.py +++ b/vllm/model_executor/layers/sparsity/__init__.py @@ -5,7 +5,7 @@ if not is_magic_wand_available: raise ValueError( "magic_wand is not available and required for sparsity " - "support. Please install it with `pip install magic_wand`") + "support. Please install it with `pip install nm-magic-wand`") from vllm.model_executor.layers.sparsity.base_config import SparsityConfig # noqa: E402 from vllm.model_executor.layers.sparsity.sparse_w16a16 import SparseW16A16Config # noqa: E402