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

更新 Dockerfile

Browse Source
This commit is contained in:
zhanghao
2025-08-28 17:27:35 +08:00
parent 2d4c9a1544
commit 967c4b7d9e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
Dockerfile
View File

@@ -3,4 +3,4 @@ FROM git.modelhub.org.cn:980/enginex-iluvatar/bi100-3.2.1-x86-ubuntu20.04-py3.10
WORKDIR /workspace WORKDIR /workspace
ENV PT_SDPA_ENABLE_HEAD_DIM_PADDING=1 ENV PT_SDPA_ENABLE_HEAD_DIM_PADDING=1
RUN pip install diffusers==0.34.0 RUN pip install diffusers==0.34.0
COPY main.py test.sh dataset.json /workspace/ COPY main.py dataset.json /workspace/