diff --git a/.ci/scripts/download_hf_hub.sh b/.ci/scripts/download_hf_hub.sh index b47fc5dd215..c0487e687c7 100644 --- a/.ci/scripts/download_hf_hub.sh +++ b/.ci/scripts/download_hf_hub.sh @@ -1,5 +1,8 @@ #!/bin/bash +# Disable HF Xet storage to avoid stalled downloads on CI runners +export HF_HUB_DISABLE_XET=1 + # Function to download files from the Hugging Face Hub # Arguments: # 1. model_id: The Hugging Face repository ID (e.g., "organization/model_name") diff --git a/.ci/scripts/export_model_artifact.sh b/.ci/scripts/export_model_artifact.sh index b7edc1765ea..1f75d850e84 100755 --- a/.ci/scripts/export_model_artifact.sh +++ b/.ci/scripts/export_model_artifact.sh @@ -67,6 +67,7 @@ if [ -z "${1:-}" ]; then exit 1 fi +# Disable HF Xet storage to avoid stalled downloads on CI runners export HF_HUB_DISABLE_XET=1 set -eux diff --git a/.ci/scripts/test_huggingface_optimum_model.py b/.ci/scripts/test_huggingface_optimum_model.py index 04ad1f5e792..7b0e69ff9b4 100644 --- a/.ci/scripts/test_huggingface_optimum_model.py +++ b/.ci/scripts/test_huggingface_optimum_model.py @@ -2,6 +2,7 @@ import gc import logging import math +import os import shutil import subprocess import tempfile @@ -9,6 +10,9 @@ from pathlib import Path from typing import List +# Disable HF Xet storage to avoid stalled downloads on CI runners +os.environ.setdefault("HF_HUB_DISABLE_XET", "1") + import torch from datasets import load_dataset diff --git a/.ci/scripts/test_lora.sh b/.ci/scripts/test_lora.sh index b20d456a673..45e4c68745a 100644 --- a/.ci/scripts/test_lora.sh +++ b/.ci/scripts/test_lora.sh @@ -6,6 +6,8 @@ # LICENSE file in the root directory of this source tree. set -exu +# Disable HF Xet storage to avoid stalled downloads on CI runners +export HF_HUB_DISABLE_XET=1 # shellcheck source=/dev/null source "$(dirname "${BASH_SOURCE[0]}")/utils.sh" diff --git a/.ci/scripts/test_lora_multimethod.sh b/.ci/scripts/test_lora_multimethod.sh index 7c468eb226b..f0b30bd4be1 100755 --- a/.ci/scripts/test_lora_multimethod.sh +++ b/.ci/scripts/test_lora_multimethod.sh @@ -6,6 +6,8 @@ # LICENSE file in the root directory of this source tree. set -exu +# Disable HF Xet storage to avoid stalled downloads on CI runners +export HF_HUB_DISABLE_XET=1 # shellcheck source=/dev/null source "$(dirname "${BASH_SOURCE[0]}")/utils.sh" diff --git a/.ci/scripts/test_phi_3_mini.sh b/.ci/scripts/test_phi_3_mini.sh index 086822bbad4..76ebb2ff3d1 100644 --- a/.ci/scripts/test_phi_3_mini.sh +++ b/.ci/scripts/test_phi_3_mini.sh @@ -6,6 +6,8 @@ # LICENSE file in the root directory of this source tree. set -exu +# Disable HF Xet storage to avoid stalled downloads on CI runners +export HF_HUB_DISABLE_XET=1 BUILD_TYPE=${1:-Debug} BUILD_DIR=${3:-cmake-out} diff --git a/.github/workflows/mlx.yml b/.github/workflows/mlx.yml index 65b8543bfd4..cdc9cd8a3d0 100644 --- a/.github/workflows/mlx.yml +++ b/.github/workflows/mlx.yml @@ -306,6 +306,8 @@ jobs: timeout: 90 script: | set -eux + # Disable HF Xet storage to avoid stalled downloads on CI runners + export HF_HUB_DISABLE_XET=1 echo "::group::Install ExecuTorch" ${CONDA_RUN} python install_executorch.py > /dev/null @@ -382,6 +384,8 @@ jobs: timeout: 90 script: | set -eux + # Disable HF Xet storage to avoid stalled downloads on CI runners + export HF_HUB_DISABLE_XET=1 echo "::group::Install ExecuTorch and configure MLX build" ${CONDA_RUN} python install_executorch.py > /dev/null @@ -510,6 +514,8 @@ jobs: timeout: 90 script: | set -eux + # Disable HF Xet storage to avoid stalled downloads on CI runners + export HF_HUB_DISABLE_XET=1 MODEL_ID="${{ matrix.model.id }}" MODEL_NAME="${{ matrix.model.name }}"