Skip to content

Commit 69e489f

Browse files
committed
fix dependencies
1 parent 9229f30 commit 69e489f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

huggingface/pytorch/training/docker/2.6/py3/cu126/Dockerfile.gpu

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -48,8 +48,8 @@ RUN pip install --no-cache-dir \
4848
trl==${TRL_VERSION} \
4949
peft==${PEFT_VERSION} \
5050
flash-attn==${FLASH_ATTN_VERSION} \
51-
dill>=0.4.0 \
52-
multiprocess>=0.70.18
51+
dill==0.4.0 \
52+
multiprocess==0.70.18
5353

5454

5555
# hf_transfer will be a built-in feature, remove the env variable then

0 commit comments

Comments
 (0)