rocm/pytorch-xdit:v25.13

Manifest digest

sha256:81954713070d67bde08595e03f62110c8a3dd66a9ae17a77d611e01f83f0f4ef

OS/ARCH

linux/amd64

Compressed size

7.07 GB

Last pushed

7 months by rocmdev

Type

Image

Manifest digest

sha256:81954713070d67bde08595e03f62110c8a3dd66a9ae17a77d611e01f83f0f4ef

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 update262.58 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.99 GB
11COPY /install/*.whl /install/ # buildkit789.25 MB
12RUN /bin/sh -c python -m873.26 MB
13COPY docker/libkineto.conf /config/libkineto.conf # buildkit476 B
14RUN /bin/sh -c echo "/opt/rocm/lib"8.41 KB
15RUN /bin/sh -c echo "/opt/rocm/lib/rocm_sysdeps/lib/"8.84 KB
16ENV PATH=/opt/rocm/bin:/opt/rocm/libexec/rocprofiler-compute:/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=a1f36ee0 B
20ARG ACCELERATE_COMMIT=b5214000 B
21ARG XDIT_COMMIT=adf26810 B
22ARG AO_COMMIT=8b722840 B
23ARG ARBITER_COMMIT=252b0301d072f9d6ea895e32f6898fed951cf2950 B
24WORKDIR /app/external116 B
25COPY ./src/patches /patches # buildkit4.33 KB
26COPY /app/external/opt_groupnorm/dist/*.whl /install/opt_groupnorm/ # buildkit136.04 KB
27RUN |7 AITER_COMMIT=de14bec0ca5a9de94e10f5cad4dc1541ac558689 LONGCONTEXTATTENTION_COMMIT=2c9b712 DIFFUSERS_COMMIT=a1f36ee139.54 KB
28RUN |7 AITER_COMMIT=de14bec0ca5a9de94e10f5cad4dc1541ac558689 LONGCONTEXTATTENTION_COMMIT=2c9b712 DIFFUSERS_COMMIT=a1f36ee34.22 MB
29RUN |7 AITER_COMMIT=de14bec0ca5a9de94e10f5cad4dc1541ac558689 LONGCONTEXTATTENTION_COMMIT=2c9b712 DIFFUSERS_COMMIT=a1f36ee501.96 MB
30RUN |7 AITER_COMMIT=de14bec0ca5a9de94e10f5cad4dc1541ac558689 LONGCONTEXTATTENTION_COMMIT=2c9b712 DIFFUSERS_COMMIT=a1f36ee7.99 MB
31RUN |7 AITER_COMMIT=de14bec0ca5a9de94e10f5cad4dc1541ac558689 LONGCONTEXTATTENTION_COMMIT=2c9b712 DIFFUSERS_COMMIT=a1f36ee117.24 MB
32RUN |7 AITER_COMMIT=de14bec0ca5a9de94e10f5cad4dc1541ac558689 LONGCONTEXTATTENTION_COMMIT=2c9b712 DIFFUSERS_COMMIT=a1f36ee8.55 MB
33RUN |7 AITER_COMMIT=de14bec0ca5a9de94e10f5cad4dc1541ac558689 LONGCONTEXTATTENTION_COMMIT=2c9b712 DIFFUSERS_COMMIT=a1f36ee73.32 MB
34RUN |7 AITER_COMMIT=de14bec0ca5a9de94e10f5cad4dc1541ac558689 LONGCONTEXTATTENTION_COMMIT=2c9b712 DIFFUSERS_COMMIT=a1f36ee150.97 MB
35RUN |7 AITER_COMMIT=de14bec0ca5a9de94e10f5cad4dc1541ac558689 LONGCONTEXTATTENTION_COMMIT=2c9b712 DIFFUSERS_COMMIT=a1f36ee110.48 MB
36RUN |7 AITER_COMMIT=de14bec0ca5a9de94e10f5cad4dc1541ac558689 LONGCONTEXTATTENTION_COMMIT=2c9b712 DIFFUSERS_COMMIT=a1f36ee136.7 MB
37RUN |7 AITER_COMMIT=de14bec0ca5a9de94e10f5cad4dc1541ac558689 LONGCONTEXTATTENTION_COMMIT=2c9b712 DIFFUSERS_COMMIT=a1f36ee77 B
38WORKDIR /app32 B
39COPY src/xDiT/ci/run.hunyuanvideo.sh /app/.ci/run.hunyuanvideo.sh # buildkit1.85 KB
40COPY /app/Hunyuanvideo /app/Hunyuanvideo # buildkit5.87 KB
41COPY src/flux/ci/run.flux.sh /app/.ci/run.flux.sh # buildkit1.94 KB
42COPY src/flux/ci/run.flux.kontext.sh /app/.ci/run.flux.kontext.sh # buildkit1.87 KB
43COPY src/flux/ci/run.flux2.sh /app/.ci/run.flux2.sh # buildkit1.85 KB
44COPY /app/Flux /app/Flux # buildkit1.01 MB
45RUN |7 AITER_COMMIT=de14bec0ca5a9de94e10f5cad4dc1541ac558689 LONGCONTEXTATTENTION_COMMIT=2c9b712 DIFFUSERS_COMMIT=a1f36ee562.52 KB
46COPY src/SD_3_5/ci/run.stablediffusion_3_5.sh /app/.ci/run.stablediffusion_3_5.sh # buildkit1.88 KB
47COPY /app/StableDiffusion3.5 /app/StableDiffusion3.5 # buildkit4.55 KB
48COPY src/wan_xdit/ci/run.wan2.1.sh /app/.ci/run.wan2.1.sh # buildkit2.15 KB
49COPY src/wan_xdit/ci/run.wan2.2.sh /app/.ci/run.wan2.2.sh # buildkit2.15 KB
50COPY /app/Wan /app/Wan # buildkit253.1 KB
51RUN |7 AITER_COMMIT=de14bec0ca5a9de94e10f5cad4dc1541ac558689 LONGCONTEXTATTENTION_COMMIT=2c9b712 DIFFUSERS_COMMIT=a1f36ee57.64 MB
52COPY /app/Z-Image /app/Z-Image # buildkit3.63 KB
53COPY src/z_image/ci/run.z_image_turbo.sh /app/.ci/run.z_image_turbo.sh # buildkit1.79 KB
54COPY .ci/run.sh /app/.ci/run.sh # buildkit602 B
55COPY .ci/quality_check.sh /app/.ci/quality_check.sh # buildkit1.52 KB
56COPY ./data/miopen/userdb /miopen_userdb # buildkit65.29 KB
57ENV MIOPEN_USER_DB_PATH=/miopen_userdb0 B
58ADD references/ /app/references/ # buildkit123 B

Command

 ARG RELEASE