primussafe/pytorch-xdit:v26.7-hyperloom1

Manifest digest

sha256:0eac1eacb46e393cadd5b6a3478c43a8c660f2ca5c9150e9cf493aa655f01537

OS/ARCH

linux/amd64

Compressed size

6.78 GB

Last pushed

about 1 month by primussafe

Type

Image

Manifest digest

sha256:0eac1eacb46e393cadd5b6a3478c43a8c660f2ca5c9150e9cf493aa655f01537

Image Layers

Image Layers

1ARG RELEASE0 B

2ARG LAUNCHPAD_BUILD_ARCH0 B

3LABEL org.opencontainers.image.version=24.040 B

4ADD file ... in / 28.36 MB

5CMD ["/bin/bash"]0 B

6ARG PYTHON_VERSION=default0 B

7ARG DEBIAN_FRONTEND=noninteractive0 B

8RUN |2 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive /bin/sh272.07 MB

9COPY /opt/python/usr /usr # buildkit32 B

10RUN |2 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive /bin/sh19.73 MB

11RUN |2 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive /bin/sh4 MB

12ENV PATH=/venv/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B

13RUN |2 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive /bin/sh6.08 MB

14COPY docker/rocm_runtime_artifacts.txt /tmp/ # buildkit737 B

15RUN |2 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive /bin/sh2.54 GB

16RUN |2 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive /bin/sh5.03 MB

17RUN |2 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive /bin/sh9.22 KB

18

ENV 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/bitcode

0 B

19ARG AITER_COMMIT=3a5261496c591b4fd7cf280d35d20dee713098fd0 B

20RUN |3 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive AITER_COMMIT=3a5261496c591b4fd7cf280d35d20dee713098fd823.95 MB

21RUN |3 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive AITER_COMMIT=3a5261496c591b4fd7cf280d35d20dee713098fd3.56 MB

22RUN |3 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive AITER_COMMIT=3a5261496c591b4fd7cf280d35d20dee713098fd8.14 MB

23COPY /ffmpeg/bin /usr/local/bin # buildkit239.68 KB

24COPY /ffmpeg/build/lib /usr/local/lib # buildkit84.71 MB

25COPY /ffmpeg/build/include /usr/local/include # buildkit289.77 KB

26COPY /ffmpeg/build/share /usr/local/share # buildkit2.98 MB

27RUN |3 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive AITER_COMMIT=3a5261496c591b4fd7cf280d35d20dee713098fd9.98 KB

28COPY docker/libkineto.conf /config/libkineto.conf # buildkit481 B

29ENV TRITON_HIP_LLD_PATH=/opt/rocm/lib/llvm/bin/ld.lld KINETO_CONFIG=/config/libkineto.conf0 B

30WORKDIR /app/external118 B

31COPY ./src/code_objects /code_objects # buildkit8.02 KB

32ENV AITER_USE_SYSTEM_TRITON=10 B

33RUN |3 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive AITER_COMMIT=3a5261496c591b4fd7cf280d35d20dee713098fd666.72 MB

34COPY docker/rocm_devel_artifacts.txt /tmp/ # buildkit882 B

35RUN |3 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive AITER_COMMIT=3a5261496c591b4fd7cf280d35d20dee713098fd1.39 GB

36RUN |3 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive AITER_COMMIT=3a5261496c591b4fd7cf280d35d20dee713098fd10.43 KB

37

ENV PATH=/opt/rocm/libexec/rocprofiler-compute:/opt/rocm/bin:/venv/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin

0 B

38COPY /venv /venv # buildkit274.9 MB

39COPY /app /app # buildkit397.13 MB

40RUN |3 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive AITER_COMMIT=3a5261496c591b4fd7cf280d35d20dee713098fd14.37 MB

41WORKDIR /app32 B

42COPY src/flux/cat.png /app/data/flux_cat.png # buildkit1 MB

43COPY src/wan_xdit/assets/i2v_input.JPG /app/data/wan_input.jpg # buildkit244.19 KB

44COPY docs/pytorch-xdit/README.md /app/README.md # buildkit817 B

45COPY .ci/run.py /app/.ci/run.py # buildkit6.65 KB

46COPY benchmark_configs/xdit/ /app/.ci/benchmark_configs/ # buildkit5.58 KB

47COPY .ci/quality_check.py /app/.ci/quality_check.py # buildkit2.43 KB

48COPY src/sweep_tooling/generate_sweep_configs.py /app/.ci/tools/generate_sweep_configs.py # buildkit1.27 KB

49COPY ./data/miopen/userdb /miopen_userdb # buildkit71.87 KB

50ENV MIOPEN_USER_DB_PATH=/miopen_userdb0 B

51ADD references/ /app/references/ # buildkit121 B

52COPY overlay_v26.7/runner_models/base_model.py 15.3 KB

53COPY overlay_v26.7/runner_models/flux.py 4.96 KB

54COPY overlay_v26.7/runner_models/stable_diffusion.py ${RUNNER_MODELS}/stable_diffusion.py2.3 KB

55COPY overlay_v26.7/core_utils/runner_utils.py 9.35 KB

56COPY overlay_v26.7/cache/registry.py 1.3 KB

57COPY overlay_v26.7/config/args.py 8.4 KB

58COPY runner_models/sana.py 1.54 KB

59COPY runner_models/stable_diffusion_xl.py ${RUNNER_MODELS}/stable_diffusion_xl.py1.66 KB

60COPY runner_models/auraflow.py 1.86 KB

61COPY transformers/transformer_auraflow.py ${TRANSFORMERS}/transformer_auraflow.py3.83 KB

62COPY runner_models/nucleus.py 2.25 KB

63COPY runner_models/hunyuan_image.py 1.86 KB

64COPY runner_models/hidream.py 1.89 KB

65COPY runner_models/ernie_image.py 1.8 KB

66COPY runner_models/ideogram4.py 5.02 KB

67COPY tools/convert_ideogram4_fp8.py 2.58 KB

68COPY diffusers_patch/diffusers_ernie_ideogram.patch /tmp/diffusers_ernie_ideogram.patch27.95 KB

69RUN cd ${DIFFUSERS} &&52.41 KB

70COPY cache/sd3.py 2.76 KB

71COPY pipelines/pipeline_stable_diffusion_3.py ${PIPELINES}/pipeline_stable_diffusion_3.py9.11 KB

72COPY runner_models/hyperloom_local_aliases.py ${RUNNER_MODELS}/hyperloom_local_aliases.py3.95 KB

73RUN ln -sf /opt/rocm/bin/hipcc 265.5 KB

74RUN "${QUALITY_PY}" -m pip install281.08 MB

75RUN find /app/xDiT/xfuser -name '__pycache__'4.53 MB

76RUN cd /app/xDiT &&21.55 MB

Command

 ARG RELEASE