Change magic_wand to nm-magic-wand (#86)

Co-authored-by: andy-neuma <[email protected]>

Author: mgoin

.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=$?

.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

.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 .

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:

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,

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