FROM zibo.harbor.iluvatar.com.cn:30000/saas/bi100-3.2.1-x86-ubuntu20.04-py3.10-poc-llm-infer:v1.2.2

WORKDIR /workspace
ENV PT_SDPA_ENABLE_HEAD_DIM_PADDING=1
RUN pip install diffusers==0.34.0
COPY main.py /workspace/
COPY test.sh /workspace/