forked from EngineX-Ascend/enginex-ascend-910-diffusers
6 lines
290 B
Docker
6 lines
290 B
Docker
FROM harbor-contest.4pd.io/zhangyiqun/public/pytorch:2.6.0-cuda12.4-cudnn9-devel
|
|
WORKDIR /workspace
|
|
# ENV PT_SDPA_ENABLE_HEAD_DIM_PADDING=1
|
|
RUN pip install diffusers transformers sentencepiece -i https://nexus.4pd.io/repository/pypi-all/simple
|
|
COPY main.py test.sh dataset.json /workspace/
|