Skip to content

Commit c93a177

Browse files
ErnevSharmashaernev
andauthored
Huggingface pytorch inference latest (#4686)
* Update ['dlc_developer_config.toml'] dlc_developer_config.toml: { 'build': { 'build_frameworks': ['huggingface_pytorch'], 'build_inference': False, 'build_training': True}, 'buildspec_override': { 'dlc-pr-huggingface-pytorch-training': 'huggingface/pytorch/training/buildspec-2-3-0.yml'}, 'dev': { 'arm64_mode': False, 'deep_canary_mode': False, 'graviton_mode': False, 'neuronx_mode': False}, 'test': { 'ec2_tests': True, 'ecs_tests': True, 'eks_tests': True, 'sagemaker_local_tests': True, 'sagemaker_remote_tests': True, 'sanity_tests': True, 'security_tests': True}} * chore: pinning cmake for huggingface pytorch training 2.3 * chore: moving patch * chore: pinning kenlm instead of cmake * chore: pinning cmake * chore: fixing syntax * Revert "Update ['dlc_developer_config.toml']" This reverts commit c0b1515. * Update ['dlc_developer_config.toml'] dlc_developer_config.toml: { 'build': { 'build_frameworks': ['huggingface_pytorch'], 'build_inference': True, 'build_training': False}, 'buildspec_override': { 'dlc-pr-huggingface-pytorch-inference': 'huggingface/pytorch/inference/buildspec.yml'}, 'dev': { 'arm64_mode': False, 'deep_canary_mode': False, 'graviton_mode': False, 'neuronx_mode': False}, 'test': { 'ec2_tests': True, 'ecs_tests': True, 'eks_tests': True, 'sagemaker_local_tests': True, 'sagemaker_remote_tests': True, 'sanity_tests': True, 'security_tests': True}} * chore: pinning cmake in conda install for huggingface pytorch inference gpu and cpu latest * chore: selectively install kenln prior to other installs * chore: pinning cmake * Revert "Update ['dlc_developer_config.toml']" This reverts commit 37a13b3. --------- Co-authored-by: shaernev <shaernev@amazon.com>
1 parent 82fcfce commit c93a177

File tree

2 files changed

+8
-2
lines changed

2 files changed

+8
-2
lines changed

huggingface/pytorch/inference/docker/2.6/py3/Dockerfile.cpu

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -118,7 +118,7 @@ RUN curl -L -o ~/miniforge3.sh https://github.com/conda-forge/miniforge/releases
118118
h5py \
119119
requests \
120120
libgcc \
121-
cmake \
121+
"cmake<=3.24.4" \
122122
packaging \
123123
"awscli<2" \
124124
boto3 \
@@ -195,6 +195,9 @@ RUN chmod +x /usr/local/bin/deep_learning_container.py
195195
# Hugging Face specific section #
196196
#################################
197197

198+
# Pre-install kenlm without build isolation so it uses system cmake
199+
RUN pip install --no-cache-dir --no-build-isolation kenlm
200+
198201
# install Hugging Face libraries and its dependencies
199202
RUN pip install --no-cache-dir \
200203
# hf_transfer will be a built-in feature, remove the extra then

huggingface/pytorch/inference/docker/2.6/py3/cu124/Dockerfile.gpu

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -168,7 +168,7 @@ RUN /opt/conda/bin/conda install -y -c conda-forge \
168168
typing \
169169
h5py \
170170
libgcc \
171-
cmake \
171+
"cmake<=3.24.4" \
172172
packaging \
173173
"awscli<2" \
174174
boto3 \
@@ -253,6 +253,9 @@ RUN chmod +x /usr/local/bin/cuda-compatibility-lib.sh
253253
# Hugging Face specific section #
254254
#################################
255255

256+
# Pre-install kenlm without build isolation so it uses system cmake
257+
RUN pip install --no-cache-dir --no-build-isolation kenlm
258+
256259
# install Hugging Face libraries and its dependencies
257260
RUN pip install --no-cache-dir \
258261
# hf_transfer will be a built-in feature, remove the extra then

0 commit comments

Comments
 (0)