Skip to content

Commit 5fb698a

Browse files
committed
dl_checkpoints: Bring back hf_transfer logic
We need it in the conatiner actaully.
1 parent 4b947e7 commit 5fb698a

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

runner/dl_checkpoints.sh

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -220,10 +220,15 @@ function run_pipeline_prepare() {
220220
# ai-worker has live-app tags hardcoded in `var livePipelineToImage` so we need to use the same tag in here
221221
docker image tag "$image" "livepeer/ai-runner:live-app-$pipeline"
222222

223+
# NOTE: We use the legacy HF_HUB_ENABLE_HF_TRANSFER env var and install hf_transfer package
224+
# because the container image has huggingface_hub==0.35.0 (from conda), which doesn't support
225+
# the newer HF_XET_HIGH_PERFORMANCE feature (introduced in v1.0+). The hf_transfer package
226+
# provides fast downloads compatible with older huggingface_hub versions.
227+
# TODO: Migrate container's huggingface_hub to v1.0+ and switch to HF_XET_HIGH_PERFORMANCE
223228
docker run --rm --name "ai-runner-${pipeline}-prepare" -v ./models:/models "${docker_run_flags[@]}" \
224229
-l "$label" \
225230
-e HF_HUB_OFFLINE=0 \
226-
-e HF_XET_HIGH_PERFORMANCE="${HF_XET_HIGH_PERFORMANCE:-1}" \
231+
-e HF_HUB_ENABLE_HF_TRANSFER=1 \
227232
-e HF_TOKEN="${HF_TOKEN:-}" \
228233
"$image" bash -c "set -euo pipefail && \
229234
$CONDA_PYTHON -m pip install --no-cache-dir hf_transfer==0.1.4 && \

0 commit comments

Comments
 (0)