From 1a006a0cf2db0531531bed5577a98daf53736c82 Mon Sep 17 00:00:00 2001 From: zhousha <736730048@qq.com> Date: Mon, 25 Aug 2025 16:22:01 +0800 Subject: [PATCH] update Dockerfile images --- Dockerfile.a100 | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Dockerfile.a100 b/Dockerfile.a100 index 7ad2ab4..cbef7b1 100644 --- a/Dockerfile.a100 +++ b/Dockerfile.a100 @@ -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