add Dockerfile for A100

Dockerfile.a100

@@ -0,0 +1,5 @@
+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/

run_in_docker_a100.sh

@@ -0,0 +1,3 @@
+#! /usr/bin/env bash
+image=harbor-contest.4pd.io/zhanghao/diffusers:a100-0.2
+docker run -it -v /home/zhanghao/workspace:/workspace -v /mnt:/mnt $image bash