Skip to content

Commit b08f346

Browse files
authored
fix: Update vLLM Dockerfile to remove the vLLM package removal (#128)
1 parent 63c4ed6 commit b08f346

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

docker/vllm/inference/0.9.1/Dockerfile.neuronx

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -223,9 +223,7 @@ RUN cd /tmp \
223223
&& git clone -b 2.25.0 https://github.com/aws-neuron/upstreaming-to-vllm.git \
224224
&& cd upstreaming-to-vllm \
225225
&& ${PIP} install --no-cache-dir -r requirements/neuron.txt \
226-
&& VLLM_TARGET_DEVICE="neuron" ${PIP} install --no-cache-dir -e . \
227-
&& cd / \
228-
&& rm -rf /tmp/upstreaming-to-vllm
226+
&& VLLM_TARGET_DEVICE="neuron" ${PIP} install --no-cache-dir -e .
229227

230228
WORKDIR ${APP_MOUNT}/vllm
231229

0 commit comments

Comments
 (0)