Skip to content

Commit 4d19ee2

Browse files
author
Bhanu Teja Goshikonda
committed
reverted package declarations for pytorch 2.8 image since tests succeeded for the fix
1 parent 35a9bf1 commit 4d19ee2

File tree

2 files changed

+2
-8
lines changed

2 files changed

+2
-8
lines changed

pytorch/training/docker/2.8/py3/Dockerfile.cpu

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -205,14 +205,11 @@ RUN pip install --no-cache-dir -U torch==${PYTORCH_VERSION} \
205205
fastai \
206206
accelerate \
207207
# pin numpy requirement for fastai dependency
208-
# requires explicit declaration of spacy, thic, blis, typer, langcodes, language_data
208+
# requires explicit declaration of spacy, thic, blis
209209
spacy \
210210
# pin thinc due to incompatibility with numpy 1.26.4 (sagemaker doesn't support latest numpy)
211211
thinc==8.3.4 \
212212
blis \
213-
typer \
214-
langcodes \
215-
language_data \
216213
numpy \
217214
&& pip uninstall -y dataclasses
218215

pytorch/training/docker/2.8/py3/cu129/Dockerfile.gpu

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -125,14 +125,11 @@ RUN pip install --no-cache-dir -U torch==${PYTORCH_VERSION} \
125125
fastai \
126126
accelerate \
127127
# pin numpy requirement for fastai dependency
128-
# requires explicit declaration of spacy, thic, blis, typer, langcodes, language_data
128+
# requires explicit declaration of spacy, thic, blis
129129
spacy \
130130
# pin thinc due to incompatibility with numpy 1.26.4 (sagemaker doesn't support latest numpy)
131131
thinc==8.3.4 \
132132
blis \
133-
typer \
134-
langcodes \
135-
language_data \
136133
numpy \
137134
&& pip uninstall -y dataclasses
138135

0 commit comments

Comments
 (0)