Unverified Commit f58eb70e authored by Daniël de Kok's avatar Daniël de Kok Committed by GitHub
Browse files

Make moe-kernels and marlin-kernels mandatory in CUDA installs (#2632)

parent 9c9ef37c
......@@ -31,7 +31,7 @@ install: install-cuda
echo "Installed server"
install-cuda: install-server install-flash-attention-v2-cuda install-vllm-cuda install-flash-attention install-fbgemm
pip install -e ".[bnb]"
pip install -e ".[bnb,marlin,moe]"
pip install nvidia-nccl-cu12==2.22.3
install-rocm: install-server install-flash-attention-v2-rocm install-vllm-rocm
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment