Skip to content

Commit 00f8fe9

Browse files
authored
Remove huggingface hub pin (#1526)
Starting with `transformers 5.0.0`, it now works with `huggingface-hub >= 1.3`
1 parent 58a688f commit 00f8fe9

2 files changed

Lines changed: 2 additions & 3 deletions

File tree

Dockerfile.tmpl

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,12 +31,11 @@ RUN uv pip install --system "tbb>=2022" "libpysal==4.9.2"
3131

3232
# b/404590350: Ray and torchtune have conflicting tune cli, we will prioritize torchtune.
3333
# b/415358158: Gensim removed from Colab image to upgrade scipy
34-
# b/456239669: remove huggingface-hub pin when pytorch-lighting and transformer are compatible
3534
# b/315753846: Unpin translate package, currently conflicts with adk 1.17.0
3635
# b/468379293: Unpin Pandas once cuml/cudf are compatible, version 3.0 causes issues
3736
# b/468383498: numpy will auto-upgrade to 2.4.x, which causes issues with numerous packages
3837
# b/468367647: Unpin protobuf, version greater than v5.29.5 causes issues with numerous packages
39-
RUN uv pip install --system --force-reinstall --no-deps torchtune gensim "scipy<=1.15.3" "huggingface-hub==0.36.0" "google-cloud-translate==3.12.1" "numpy==2.0.2" "pandas==2.2.2"
38+
RUN uv pip install --system --force-reinstall --no-deps torchtune gensim "scipy<=1.15.3" "google-cloud-translate==3.12.1" "numpy==2.0.2" "pandas==2.2.2"
4039
RUN uv pip install --system --force-reinstall "protobuf==5.29.5"
4140

4241
# Adding non-package dependencies:

kaggle_requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -123,7 +123,7 @@ tensorflow-io
123123
torchinfo
124124
torchmetrics
125125
torchtune
126-
transformers>=4.51.0
126+
transformers>=5.0.0
127127
vtk
128128
wavio
129129
xvfbwrapper

0 commit comments

Comments
 (0)