Sign inSign up

rocm/pytorch-xdit:v25.12

Manifest digest

sha256:e06895132316bf3c393366b70a91eaab6755902dad0100e6e2b38310547d9256

OS/ARCH

linux/amd64

Compressed size

6.55 GB

Last pushed

7 months by rocmdev

Type

Image

Manifest digest

sha256:e06895132316bf3c393366b70a91eaab6755902dad0100e6e2b38310547d9256

Image Layers

Image Layers

1ARG RELEASE0 B
2ARG LAUNCHPAD_BUILD_ARCH0 B
3LABEL org.opencontainers.image.ref.name=ubuntu0 B
4LABEL org.opencontainers.image.version=24.040 B
5ADD file ... in / 28.35 MB
6CMD ["/bin/bash"]0 B
7RUN /bin/sh -c apt-get update259.92 MB
8RUN /bin/sh -c python3 -m4 MB
9ENV PATH=/venv/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
10COPY /opt/rocm /opt/rocm # buildkit3.64 GB
11COPY /install/*.whl /install/ # buildkit784.17 MB
12RUN /bin/sh -c python -m868.47 MB
13COPY docker/libkineto.conf /config/libkineto.conf # buildkit483 B
14RUN /bin/sh -c echo "/opt/rocm/lib"8.05 KB
15RUN /bin/sh -c echo "/opt/rocm/lib/rocm_sysdeps/lib/"8.4 KB
16ENV PATH=/opt/rocm/bin:/venv/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin ROCM_HOME=/opt/rocm ROCM_PATH=/opt/rocm HIP_DEVICE_LIB_PATH=/opt/rocm/lib/llvm/amdgcn/bitcode0 B
17ARG AITER_COMMIT=de14bec0ca5a9de94e10f5cad4dc1541ac5586890 B
18ARG LONGCONTEXTATTENTION_COMMIT=2c9b7120 B
19ARG DIFFUSERS_COMMIT=40528e90 B
20ARG ACCELERATE_COMMIT=b5214000 B
21ARG XDIT_COMMIT=ccba9d50 B
22ARG AO_COMMIT=8b722840 B
23WORKDIR /app/external117 B
24COPY ./src/patches /patches # buildkit2.5 KB
25COPY /app/external/opt_groupnorm/dist/*.whl /install/opt_groupnorm/ # buildkit138.15 KB
26RUN |6 AITER_COMMIT=de14bec0ca5a9de94e10f5cad4dc1541ac558689 LONGCONTEXTATTENTION_COMMIT=2c9b712 DIFFUSERS_COMMIT=40528e9141.64 KB
27RUN |6 AITER_COMMIT=de14bec0ca5a9de94e10f5cad4dc1541ac558689 LONGCONTEXTATTENTION_COMMIT=2c9b712 DIFFUSERS_COMMIT=40528e934.2 MB
28RUN |6 AITER_COMMIT=de14bec0ca5a9de94e10f5cad4dc1541ac558689 LONGCONTEXTATTENTION_COMMIT=2c9b712 DIFFUSERS_COMMIT=40528e9499.58 MB
29RUN |6 AITER_COMMIT=de14bec0ca5a9de94e10f5cad4dc1541ac558689 LONGCONTEXTATTENTION_COMMIT=2c9b712 DIFFUSERS_COMMIT=40528e97.99 MB
30RUN |6 AITER_COMMIT=de14bec0ca5a9de94e10f5cad4dc1541ac558689 LONGCONTEXTATTENTION_COMMIT=2c9b712 DIFFUSERS_COMMIT=40528e9115.86 MB
31RUN |6 AITER_COMMIT=de14bec0ca5a9de94e10f5cad4dc1541ac558689 LONGCONTEXTATTENTION_COMMIT=2c9b712 DIFFUSERS_COMMIT=40528e98.49 MB
32RUN |6 AITER_COMMIT=de14bec0ca5a9de94e10f5cad4dc1541ac558689 LONGCONTEXTATTENTION_COMMIT=2c9b712 DIFFUSERS_COMMIT=40528e973.27 MB
33RUN |6 AITER_COMMIT=de14bec0ca5a9de94e10f5cad4dc1541ac558689 LONGCONTEXTATTENTION_COMMIT=2c9b712 DIFFUSERS_COMMIT=40528e9148.71 MB
34RUN |6 AITER_COMMIT=de14bec0ca5a9de94e10f5cad4dc1541ac558689 LONGCONTEXTATTENTION_COMMIT=2c9b712 DIFFUSERS_COMMIT=40528e997 B
35WORKDIR /app32 B
36COPY src/xDiT/ci/run.hunyuanvideo.sh /app/.ci/run.hunyuanvideo.sh # buildkit1.84 KB
37COPY /app/Hunyuanvideo /app/Hunyuanvideo # buildkit5.87 KB
38COPY src/flux/ci/run.flux.sh /app/.ci/run.flux.sh # buildkit1.93 KB
39COPY /app/Flux /app/Flux # buildkit5.05 KB
40RUN |6 AITER_COMMIT=de14bec0ca5a9de94e10f5cad4dc1541ac558689 LONGCONTEXTATTENTION_COMMIT=2c9b712 DIFFUSERS_COMMIT=40528e9562.43 KB
41COPY src/SD_3_5/ci/run.stablediffusion_3_5.sh /app/.ci/run.stablediffusion_3_5.sh # buildkit1.88 KB
42COPY /app/StableDiffusion3.5 /app/StableDiffusion3.5 # buildkit4.55 KB
43COPY src/wan_xdit/ci/run.wan2.1.sh /app/.ci/run.wan2.1.sh # buildkit2.13 KB
44COPY src/wan_xdit/ci/run.wan2.2.sh /app/.ci/run.wan2.2.sh # buildkit2.12 KB
45COPY /app/Wan /app/Wan # buildkit252.59 KB
46RUN |6 AITER_COMMIT=de14bec0ca5a9de94e10f5cad4dc1541ac558689 LONGCONTEXTATTENTION_COMMIT=2c9b712 DIFFUSERS_COMMIT=40528e9144.77 MB
47COPY .ci/run.sh /app/.ci/run.sh # buildkit551 B
48COPY ./data/miopen/userdb /miopen_userdb # buildkit64.88 KB
49ENV MIOPEN_USER_DB_PATH=/miopen_userdb0 B

Command

 ARG RELEASE