更新 Dockerfile

This commit is contained in:
2025-08-28 17:27:35 +08:00
parent 2d4c9a1544
commit 967c4b7d9e

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
ENV PT_SDPA_ENABLE_HEAD_DIM_PADDING=1
RUN pip install diffusers==0.34.0
COPY main.py test.sh dataset.json /workspace/
COPY main.py dataset.json /workspace/