mirror of
https://github.com/kvcache-ai/ktransformers.git
synced 2026-03-14 18:37:23 +00:00
171578a7ec
* Change named 'KT-SFT' to 'kt-sft' * [docs]: update kt-sft name --------- Co-authored-by: ZiWei Yuan <yzwliam@126.com>
27 lines
959 B
Bash
Executable File
27 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"
|