Skip to content

Commit

Permalink
[docker] pin tokenizers version in neuronx
Browse files Browse the repository at this point in the history
  • Loading branch information
sindhuvahinis committed Jan 17, 2025
1 parent 36eaa27 commit 7694ce6
Showing 1 changed file with 5 additions and 2 deletions.
7 changes: 5 additions & 2 deletions serving/docker/pytorch-inf2.Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -30,6 +30,7 @@ ARG torch_xla_version=2.1.6
ARG libneuronxla_version=2.1.681.0

ARG transformers_version=4.45.2
ARG tokenizers_version=0.20.3
ARG accelerate_version=0.29.2
ARG diffusers_version=0.28.2
ARG pydantic_version=2.6.1
Expand Down Expand Up @@ -107,14 +108,16 @@ RUN mkdir -p /opt/djl/bin && cp scripts/telemetry.sh /opt/djl/bin && \
pydantic==${pydantic_version} \
optimum \
tiktoken \
blobfile && \
blobfile \
msgspec && \
# Install packages with no-deps flag
pip install --no-deps \
neuronx_distributed_inference==${neuronx_distributed_inference_version} \
optimum-neuron==${optimum_neuron_version} \
transformers-neuronx==${transformers_neuronx_version} \
${vllm_wheel} \
transformers==${transformers_version} && \
transformers==${transformers_version} \
tokenizers==${tokenizers_version} && \
# Install s5cmd and patch OSS DLC
scripts/install_s5cmd.sh x64 && \
scripts/patch_oss_dlc.sh python && \
Expand Down

0 comments on commit 7694ce6

Please sign in to comment.