Skip to content

Commit 5d36786

Browse files
committed
Fix Hugging Face dependency pins for CUDA jobs
1 parent 54af057 commit 5d36786

3 files changed

Lines changed: 4 additions & 4 deletions

File tree

.github/workflows/cuda-perf.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -124,7 +124,7 @@ jobs:
124124
echo "::endgroup::"
125125
126126
echo "::group::Setup Huggingface"
127-
pip install -U "huggingface_hub[cli]<1.0" accelerate "optimum~=2.0.0" "transformers==5.0.0rc1"
127+
pip install -U "huggingface_hub[cli]>=1.2.1,<2.0" accelerate "optimum~=2.0.0" "transformers==5.0.0rc1"
128128
huggingface-cli login --token $SECRET_EXECUTORCH_HF_TOKEN
129129
OPTIMUM_ET_VERSION=$(cat .ci/docker/ci_commit_pins/optimum-executorch.txt)
130130
pip install --no-deps git+https://github.com/huggingface/optimum-executorch.git@${OPTIMUM_ET_VERSION}

.github/workflows/cuda-windows.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -98,7 +98,7 @@ jobs:
9898
# Setup Huggingface only for models that need it (not dinov2)
9999
if [ "${{ matrix.model_name }}" != "dinov2-small-imagenet1k-1-layer" ]; then
100100
echo "::group::Setup Huggingface"
101-
pip install -U "huggingface_hub[cli]<1.0" accelerate "optimum~=2.0.0" "transformers==5.0.0rc1"
101+
pip install -U "huggingface_hub[cli]>=1.2.1,<2.0" accelerate "optimum~=2.0.0" "transformers==5.0.0rc1"
102102
huggingface-cli login --token $SECRET_EXECUTORCH_HF_TOKEN
103103
OPTIMUM_ET_VERSION=$(cat .ci/docker/ci_commit_pins/optimum-executorch.txt)
104104
pip install --no-deps git+https://github.com/huggingface/optimum-executorch.git@${OPTIMUM_ET_VERSION}

.github/workflows/cuda.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -288,7 +288,7 @@ jobs:
288288
# Setup Huggingface only for models that need it (not parakeet or dinov2)
289289
if [ "${{ matrix.model.name }}" != "parakeet-tdt" ] && [ "${{ matrix.model.name }}" != "dinov2-small-imagenet1k-1-layer" ]; then
290290
echo "::group::Setup Huggingface"
291-
pip install -U "huggingface_hub[cli]<1.0" accelerate "optimum~=2.0.0" "transformers==5.0.0rc1"
291+
pip install -U "huggingface_hub[cli]>=1.2.1,<2.0" accelerate "optimum~=2.0.0" "transformers==5.0.0rc1"
292292
huggingface-cli login --token $SECRET_EXECUTORCH_HF_TOKEN
293293
OPTIMUM_ET_VERSION=$(cat .ci/docker/ci_commit_pins/optimum-executorch.txt)
294294
pip install --no-deps git+https://github.com/huggingface/optimum-executorch.git@${OPTIMUM_ET_VERSION}
@@ -469,7 +469,7 @@ jobs:
469469
echo "::endgroup::"
470470
471471
echo "::group::Setup Huggingface"
472-
pip install -U "huggingface_hub[cli]<1.0"
472+
pip install -U "huggingface_hub[cli]>=1.2.1,<2.0"
473473
huggingface-cli login --token $SECRET_EXECUTORCH_HF_TOKEN
474474
echo "::endgroup::"
475475

0 commit comments

Comments
 (0)