FROM cr.metax-tech.com/public-ai-release/maca/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

RUN /opt/conda/bin/pip install easydict dashscope

WORKDIR /opt/app

COPY ./main.py ./dataset.json ./wan_pipeline.py ./
COPY ./Wan2.1 ./Wan2.1

ENTRYPOINT ["/opt/conda/bin/python"]