7 lines
240 B
Docker
7 lines
240 B
Docker
FROM git.modelhub.org.cn:980/enginex-iluvatar/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/
|