7 lines
238 B
Docker
7 lines
238 B
Docker
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 test.sh dataset.json /workspace/
|