File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change 11# default base image
22# https://gallery.ecr.aws/neuron/pytorch-inference-neuronx
3- ARG BASE_IMAGE="public.ecr.aws/neuron/pytorch-inference-neuronx:2.5.1-neuronx-py310-sdk2.21 .0-ubuntu22.04"
3+ ARG BASE_IMAGE="public.ecr.aws/neuron/pytorch-inference-neuronx:2.5.1-neuronx-py310-sdk2.22 .0-ubuntu22.04"
44
55FROM $BASE_IMAGE
66
@@ -21,9 +21,9 @@ VOLUME [ ${APP_MOUNT} ]
2121WORKDIR ${APP_MOUNT}/vllm
2222
2323RUN python3 -m pip install --upgrade pip
24- RUN python3 -m pip install --no-cache-dir fastapi ninja tokenizers pandas
25- RUN python3 -m pip install sentencepiece transformers==4.45.2 -U
26- RUN python3 -m pip install neuronx-cc==2.16.345 .0 --extra-index-url=https://pip.repos.neuron.amazonaws.com -U
24+ RUN python3 -m pip install --no-cache-dir fastapi ninja tokenizers pandas tenacity
25+ RUN python3 -m pip install sentencepiece transformers==4.48.0 -U
26+ RUN python3 -m pip install neuronx-cc==2.17.194 .0 --extra-index-url=https://pip.repos.neuron.amazonaws.com -U
2727RUN python3 -m pip install pytest
2828
2929# uninstall transformers-neuronx package explicitly to avoid version conflict
You can’t perform that action at this time.
0 commit comments