Skip to content

Commit 523ec52

Browse files
authored
Pin huggingface-hub (#1510)
pytorch-lighting and transformer are throwing errors due to huggingface-hub requirements. let's pin huggingface-hub for now
1 parent 0d57d8d commit 523ec52

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

Dockerfile.tmpl

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,8 @@ RUN uv pip install --system "tbb>=2022" "libpysal==4.9.2"
4848

4949
# b/404590350: Ray and torchtune have conflicting tune cli, we will prioritize torchtune.
5050
# b/415358158: Gensim removed from Colab image to upgrade scipy
51-
RUN uv pip install --system --force-reinstall --no-deps torchtune gensim "scipy<=1.15.3"
51+
# b/456239669: remove huggingface-hub pin when pytorch-lighting and transformer are compatible
52+
RUN uv pip install --system --force-reinstall --no-deps torchtune gensim "scipy<=1.15.3" "huggingface-hub==0.36.0"
5253

5354
# Adding non-package dependencies:
5455
ADD clean-layer.sh /tmp/clean-layer.sh

0 commit comments

Comments
 (0)