2025-08-25 16:29:37 +08:00
|
|
|
FROM git.modelhub.org.cn:980/enginex-iluvatar/a100-pytorch:2.6.0-cuda12.4-cudnn9-devel
|
2025-08-21 11:00:53 +08:00
|
|
|
WORKDIR /workspace
|
|
|
|
|
# ENV PT_SDPA_ENABLE_HEAD_DIM_PADDING=1
|
2025-08-26 14:48:12 +08:00
|
|
|
RUN pip install diffusers==0.34.0 sentencepiece transformers==4.55.2
|
2025-08-21 11:00:53 +08:00
|
|
|
COPY main.py test.sh dataset.json /workspace/
|