File tree Expand file tree Collapse file tree 1 file changed +6
-2
lines changed
huggingface/pytorch/training/docker/2.1/py3/cu121 Expand file tree Collapse file tree 1 file changed +6
-2
lines changed Original file line number Diff line number Diff line change @@ -19,6 +19,8 @@ ARG FLASH_ATTN_VERSION=2.3.6
19
19
ARG MULTIPROCESS_VERSION=0.70.16
20
20
ARG DILL_VERSION=0.3.8
21
21
ARG NINJA_VERSION=1.11.1
22
+ ARG HUGGINGFACE_HUB_VERSION=0.25.2
23
+ ARG GEVENT_VERSION=24.10.1
22
24
ARG PYTHON=python3
23
25
24
26
# TODO: Remove when the base image is updated
@@ -42,11 +44,13 @@ RUN pip install --no-cache-dir \
42
44
flash-attn==${FLASH_ATTN_VERSION} \
43
45
multiprocess==${MULTIPROCESS_VERSION} \
44
46
dill==${DILL_VERSION} \
45
- ninja==${NINJA_VERSION}
47
+ ninja==${NINJA_VERSION} \
48
+ huggingface_hub==${HUGGINGFACE_HUB_VERSION} \
49
+ gevent==${GEVENT_VERSION}
46
50
47
51
RUN apt-get update \
48
52
# TODO: Remove upgrade statements once packages are updated in base image
49
- && apt-get -y upgrade --only-upgrade systemd openssl cryptsetup libkrb5-3 \
53
+ && apt-get -y upgrade --only-upgrade systemd openssl cryptsetup libkrb5-3 expat libarchive13 apparmor \
50
54
&& apt install -y git git-lfs \
51
55
&& apt-get clean \
52
56
&& rm -rf /var/lib/apt/lists/*
You can’t perform that action at this time.
0 commit comments