sunjichen/enginex-ascend-910-diffusers
1
0
Fork 0
forked from EngineX-Ascend/enginex-ascend-910-diffusers
Code Pull Requests Activity

fix diffusers version

Browse Source
This commit is contained in:
ZHANG Hao
2025-08-28 11:17:44 +08:00
parent be947dfc79
commit cdefc1873e
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 pytorch/pytorch:2.6.0-cuda12.4-cudnn9-devel FROM pytorch/pytorch:2.6.0-cuda12.4-cudnn9-devel
WORKDIR /workspace WORKDIR /workspace
# ENV PT_SDPA_ENABLE_HEAD_DIM_PADDING=1 # ENV PT_SDPA_ENABLE_HEAD_DIM_PADDING=1
RUN pip install diffusers transformers sentencepiece RUN pip install diffusers==0.34.0 transformers sentencepiece
COPY main.py test.sh dataset.json /workspace/ COPY main.py test.sh dataset.json /workspace/