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/