Skip to content

Commit f645cdb

Browse files
authored
add no deps to peft
1 parent 1efc5ac commit f645cdb

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

huggingface/pytorch/inference/docker/2.7/py3/sdk2.24.1/Dockerfile.neuronx

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -173,9 +173,9 @@ RUN pip install --no-cache-dir \
173173
hf_transfer \
174174
datasets==${DATASETS_VERSION} \
175175
optimum-neuron==${OPTIMUM_NEURON_VERSION} \
176-
peft==${PEFT_VERSION} \
177176
"sagemaker-huggingface-inference-toolkit>=2.4.1,<3" \
178177
&& pip install --no-deps --no-cache-dir -U sentence_transformers==${SENTENCE_TRANSFORMERS} \
178+
peft==${PEFT_VERSION} \
179179
&& rm -rf ~/.cache/pip/*
180180

181181
RUN HOME_DIR=/root \
@@ -192,4 +192,4 @@ RUN HOME_DIR=/root \
192192
ENV HF_HUB_USER_AGENT_ORIGIN="aws:sagemaker:neuron:inference:regular"
193193
EXPOSE 8080 8081
194194
ENTRYPOINT ["python", "/usr/local/bin/dockerd-entrypoint.py"]
195-
CMD ["serve"]
195+
CMD ["serve"]

0 commit comments

Comments
 (0)