[docker][inf2] pin huf hub python dependency (#2538)

deepjavalibrary · Nov 8, 2024 · 17ac121 · 17ac121
1 parent 87b86bb
commit 17ac121
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/serving/docker/pytorch-inf2.Dockerfile b/serving/docker/pytorch-inf2.Dockerfile
@@ -77,7 +77,7 @@ RUN mkdir -p /opt/djl/bin && cp scripts/telemetry.sh /opt/djl/bin && \
     neuronx-cc==${neuronx_cc_version} torch-neuronx==${torch_neuronx_version} transformers-neuronx==${transformers_neuronx_version} \
     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 \
+    pydantic==${pydantic_version} optimum optimum-neuron==${optimum_neuron_version} tiktoken blobfile "huggingface_hub<0.26" \
     torchvision==${torchvision_version} && \
     pip install numpy==${numpy_version} && \
     scripts/install_s5cmd.sh x64 && \