diff --git a/Dockerfile b/Dockerfile index 364c3c9..1bcdccf 100644 --- a/Dockerfile +++ b/Dockerfile @@ -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/