This commit is contained in:
Alisehen 2025-05-15 07:28:50 +00:00
parent 055680e26c
commit edd9efa49e

View file

@ -28,6 +28,9 @@ echo "Installing python dependencies from requirements.txt"
pip install -r requirements-local_chat.txt
pip install -r ktransformers/server/requirements.txt
echo "Installing ktransformers"
KTRANSFORMERS_FORCE_BUILD=TRUE pip install -v . --no-build-isolation
# XPU-specific fix for triton
if [[ "$DEV_BACKEND" == "xpu" ]]; then
echo "Replacing triton for XPU backend"
@ -35,9 +38,6 @@ if [[ "$DEV_BACKEND" == "xpu" ]]; then
pip install pytorch-triton-xpu==3.3.0 --extra-index-url https://download.pytorch.org/whl/xpu
fi
echo "Installing ktransformers"
KTRANSFORMERS_FORCE_BUILD=TRUE pip install -v . --no-build-isolation
if [[ "$DEV_BACKEND" == "cuda" ]]; then
echo "Installing custom_flashinfer for CUDA backend"
pip install third_party/custom_flashinfer/