File tree Expand file tree Collapse file tree 1 file changed +6
-1
lines changed Expand file tree Collapse file tree 1 file changed +6
-1
lines changed Original file line number Diff line number Diff 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 && \
You can’t perform that action at this time.
0 commit comments