File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 11#! /bin/bash
22
3+ # Disable HF Xet storage to avoid stalled downloads on CI runners
4+ export HF_HUB_DISABLE_XET=1
5+
36# Function to download files from the Hugging Face Hub
47# Arguments:
58# 1. model_id: The Hugging Face repository ID (e.g., "organization/model_name")
Original file line number Diff line number Diff line change @@ -67,6 +67,7 @@ if [ -z "${1:-}" ]; then
6767 exit 1
6868fi
6969
70+ # Disable HF Xet storage to avoid stalled downloads on CI runners
7071export HF_HUB_DISABLE_XET=1
7172
7273set -eux
Original file line number Diff line number Diff line change 22import gc
33import logging
44import math
5+ import os
56import shutil
67import subprocess
78import tempfile
89import time
910from pathlib import Path
1011from typing import List
1112
13+ # Disable HF Xet storage to avoid stalled downloads on CI runners
14+ os .environ .setdefault ("HF_HUB_DISABLE_XET" , "1" )
15+
1216import torch
1317from datasets import load_dataset
1418
Original file line number Diff line number Diff line change 66# LICENSE file in the root directory of this source tree.
77
88set -exu
9+ # Disable HF Xet storage to avoid stalled downloads on CI runners
10+ export HF_HUB_DISABLE_XET=1
911# shellcheck source=/dev/null
1012source " $( dirname " ${BASH_SOURCE[0]} " ) /utils.sh"
1113
Original file line number Diff line number Diff line change 66# LICENSE file in the root directory of this source tree.
77
88set -exu
9+ # Disable HF Xet storage to avoid stalled downloads on CI runners
10+ export HF_HUB_DISABLE_XET=1
911# shellcheck source=/dev/null
1012source " $( dirname " ${BASH_SOURCE[0]} " ) /utils.sh"
1113
Original file line number Diff line number Diff line change 66# LICENSE file in the root directory of this source tree.
77
88set -exu
9+ # Disable HF Xet storage to avoid stalled downloads on CI runners
10+ export HF_HUB_DISABLE_XET=1
911
1012BUILD_TYPE=${1:- Debug}
1113BUILD_DIR=${3:- cmake-out}
Original file line number Diff line number Diff line change @@ -306,6 +306,8 @@ jobs:
306306 timeout : 90
307307 script : |
308308 set -eux
309+ # Disable HF Xet storage to avoid stalled downloads on CI runners
310+ export HF_HUB_DISABLE_XET=1
309311
310312 echo "::group::Install ExecuTorch"
311313 ${CONDA_RUN} python install_executorch.py > /dev/null
@@ -382,6 +384,8 @@ jobs:
382384 timeout : 90
383385 script : |
384386 set -eux
387+ # Disable HF Xet storage to avoid stalled downloads on CI runners
388+ export HF_HUB_DISABLE_XET=1
385389
386390 echo "::group::Install ExecuTorch and configure MLX build"
387391 ${CONDA_RUN} python install_executorch.py > /dev/null
@@ -510,6 +514,8 @@ jobs:
510514 timeout : 90
511515 script : |
512516 set -eux
517+ # Disable HF Xet storage to avoid stalled downloads on CI runners
518+ export HF_HUB_DISABLE_XET=1
513519
514520 MODEL_ID="${{ matrix.model.id }}"
515521 MODEL_NAME="${{ matrix.model.name }}"
You can’t perform that action at this time.
0 commit comments