From 8e9a6e327809d6d62a1da886fdc277701c17d0ad Mon Sep 17 00:00:00 2001
From: zhousha <736730048@qq.com>
Date: Tue, 26 Aug 2025 14:48:12 +0800
Subject: [PATCH] update Dockerfile.a100

---
 Dockerfile.a100 | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Dockerfile.a100 b/Dockerfile.a100
index cbef7b1..b821f91 100644
--- a/Dockerfile.a100
+++ b/Dockerfile.a100
@@ -1,5 +1,5 @@
 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
+RUN pip install diffusers==0.34.0 sentencepiece transformers==4.55.2
 COPY main.py test.sh dataset.json /workspace/