Files
ray-project--ray/ci/compile_llm_requirements.sh
2026-07-13 13:17:40 +08:00

17 lines
533 B
Bash
Executable File

#!/bin/bash
set -euo pipefail
CONFIG_PATH="${1:-ci/raydepsets/configs/rayllm.depsets.yaml}"
mkdir -p /tmp/ray-deps
# Remove the GPU constraints
cp python/requirements_compiled.txt /tmp/ray-deps/requirements_compiled.txt
sed -e '/^--extra-index-url /d' -e '/^--find-links /d' /tmp/ray-deps/requirements_compiled.txt > /tmp/ray-deps/requirements_compiled.txt.tmp
mv /tmp/ray-deps/requirements_compiled.txt.tmp /tmp/ray-deps/requirements_compiled.txt
bazel run //ci/raydepsets:raydepsets -- build "${CONFIG_PATH}"
echo "--- Done"