10 lines
407 B
Docker
10 lines
407 B
Docker
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
|
|
|
|
WORKDIR /opt/app
|
|
|
|
COPY ./main.py ./dataset.json ./
|
|
|
|
ENTRYPOINT ["/opt/conda/bin/python"]
|