Skip to content

Commit 2cdda2f

Browse files
committed
fix pathos dependencies
1 parent d1c8e9e commit 2cdda2f

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

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

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,8 @@ RUN pip install --no-cache-dir \
4949
peft==${PEFT_VERSION} \
5050
flash-attn==${FLASH_ATTN_VERSION} \
5151
dill==0.3.8 \
52-
multiprocess==0.70.16
52+
multiprocess==0.70.16 \
53+
pathos==0.3.3
5354

5455
# hf_transfer will be a built-in feature, remove the env variable then
5556
ENV HF_HUB_ENABLE_HF_TRANSFER="1"

0 commit comments

Comments
 (0)