Download kernels in install-cuda target

This commit is contained in:
Daniël de Kok 2025-03-10 14:36:58 +00:00
parent 141c97e34e
commit c6dd5d0e22

View File

@ -39,6 +39,7 @@ install: install-cuda
install-cuda: install-server install-flash-attention-v2-cuda install-flash-attention install-cuda: install-server install-flash-attention-v2-cuda install-flash-attention
uv pip install -e ".[attention,bnb,marlin,moe]" uv pip install -e ".[attention,bnb,marlin,moe]"
uv pip install nvidia-nccl-cu12==2.22.3 uv pip install nvidia-nccl-cu12==2.22.3
kernels download .
install-rocm: install-server install-flash-attention-v2-rocm install-vllm-rocm install-rocm: install-server install-flash-attention-v2-rocm install-vllm-rocm