update Dockerfile images

This commit is contained in:
zhousha
2025-08-25 16:22:01 +08:00
parent 9f58f90d97
commit 1a006a0cf2

View File

@@ -1,4 +1,4 @@
FROM git.modelhub.org.cn:980/enginex-iluvatar/mr100-pytorch:2.6.0-cuda12.4-cudnn9-devel
FROM git.modelhub.org.cn:980/enginex-iluvatar/a100-pytorch:2.6.0-cuda12.4-cudnn9-devel
WORKDIR /workspace
# ENV PT_SDPA_ENABLE_HEAD_DIM_PADDING=1
RUN pip install diffusers transformers sentencepiece -i https://nexus.4pd.io/repository/pypi-all/simple