init commit

This commit is contained in:
qiliguo
2025-09-16 18:23:40 +08:00
commit 4ec3a87b14
4 changed files with 198 additions and 0 deletions

9
Dockerfile Normal file
View File

@@ -0,0 +1,9 @@
FROM git.modelhub.org.cn:9443/enginex-metax/diffusers.training:maca.ai3.0.0.5-torch2.4-py310-ubuntu22.04-amd64
RUN /opt/conda/bin/pip install pytorch_lightning opencv-python-headless==4.10.0.84 imageio[ffmpeg] einops datasets==3.2.0 simplejson open_clip_torch==2.24.0 sortedcontainers modelscope av==11.0.0 addict
WORKDIR /opt/app
COPY ./main.py ./dataset.json ./
ENTRYPOINT ["/opt/conda/bin/python"]