EngineX-Iluvatar/enginex-bi_series-diffusers
15
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

update Dockerfile.a100

Browse Source
This commit is contained in:
zhousha
2025-08-26 14:48:12 +08:00
parent c11ad090e4
commit 8e9a6e3278
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
Dockerfile.a100
View File

@@ -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/