feat: support wan2.1
This commit is contained in:
@@ -2,8 +2,11 @@ FROM cr.metax-tech.com/public-ai-release/maca/diffusers.training:maca.ai3.0.0.5-
|
||||
|
||||
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 ./
|
||||
COPY ./main.py ./dataset.json ./wan_pipeline.py ./
|
||||
COPY ./Wan2.1 ./Wan2.1
|
||||
|
||||
ENTRYPOINT ["/opt/conda/bin/python"]
|
||||
|
||||
Reference in New Issue
Block a user