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
1919ARG MULTIPROCESS_VERSION=0.70.16
2020ARG DILL_VERSION=0.3.8
2121ARG NINJA_VERSION=1.11.1
22+ ARG HUGGINGFACE_HUB_VERSION=0.25.2
23+ ARG GEVENT_VERSION=24.10.1
2224ARG PYTHON=python3
2325
2426# TODO: Remove when the base image is updated
@@ -42,11 +44,13 @@ RUN pip install --no-cache-dir \
4244 flash-attn==${FLASH_ATTN_VERSION} \
4345 multiprocess==${MULTIPROCESS_VERSION} \
4446 dill==${DILL_VERSION} \
45- ninja==${NINJA_VERSION}
47+ ninja==${NINJA_VERSION} \
48+ huggingface_hub==${HUGGINGFACE_HUB_VERSION} \
49+ gevent==${GEVENT_VERSION}
4650
4751RUN apt-get update \
4852 # 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 \
5054 && apt install -y git git-lfs \
5155 && apt-get clean \
5256 && rm -rf /var/lib/apt/lists/*
You can’t perform that action at this time.
0 commit comments