mirror of
https://github.com/kvcache-ai/ktransformers.git
synced 2025-09-05 12:09:48 +00:00
26 lines
959 B
Bash
Executable file
26 lines
959 B
Bash
Executable file
#!/bin/bash
|
|
set -e
|
|
|
|
# clear build dirs
|
|
# rm -rf build
|
|
# rm -rf *.egg-info
|
|
# rm -rf csrc/build
|
|
# rm -rf csrc/ktransformers_ext/build
|
|
# rm -rf csrc/ktransformers_ext/cuda/build
|
|
# rm -rf csrc/ktransformers_ext/cuda/dist
|
|
# rm -rf csrc/ktransformers_ext/cuda/*.egg-info
|
|
rm -rf ~/.ktransformers
|
|
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 USE_BALANCE_SERVE=1 pip install -v . --no-build-isolation
|
|
pip install third_party/custom_flashinfer/ -v
|
|
|
|
# SITE_PACKAGES=$(python -c "import site; print(site.getsitepackages()[0])")
|
|
# echo "Copying thirdparty libs to $SITE_PACKAGES"
|
|
# cp -a csrc/balance_serve/build/third_party/prometheus-cpp/lib/libprometheus-cpp-*.so* $SITE_PACKAGES/
|
|
# patchelf --set-rpath '$ORIGIN' $SITE_PACKAGES/sched_ext.cpython*
|
|
|
|
|
|
echo "Installation completed successfully"
|