[docker][inf2] pin hf hub python dependency (#2537)

deepjavalibrary · Nov 8, 2024 · d7729ce · d7729ce
1 parent 3d43751
commit d7729ce
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/serving/docker/pytorch-inf2.Dockerfile b/serving/docker/pytorch-inf2.Dockerfile
@@ -78,7 +78,7 @@ RUN mkdir -p /opt/djl/bin && cp scripts/telemetry.sh /opt/djl/bin && \
     neuronx_distributed==${neuronx_distributed_version} protobuf==${protobuf_version} sentencepiece jinja2 \
     diffusers==${diffusers_version} opencv-contrib-python-headless  Pillow --extra-index-url=https://pip.repos.neuron.amazonaws.com \
     pydantic==${pydantic_version} optimum optimum-neuron==${optimum_neuron_version} tiktoken blobfile && \
-    pip install transformers==${transformers_version} && \
+    pip install transformers==${transformers_version} "huggingface_hub<0.26" && \
     scripts/install_s5cmd.sh x64 && \
     scripts/patch_oss_dlc.sh python && \
     useradd -m -d /home/djl djl && \