From ff08a69acb3fbd99d4a9abeb9c1f94677a7c3f4c Mon Sep 17 00:00:00 2001 From: michelle-yooh Date: Mon, 23 Feb 2026 21:12:32 +0000 Subject: [PATCH] Remove GPU image build steps from workflow and update README to clarify GPU support status --- .github/workflows/UploadDockerImages.yml | 13 ------------- README.md | 2 ++ 2 files changed, 2 insertions(+), 13 deletions(-) diff --git a/.github/workflows/UploadDockerImages.yml b/.github/workflows/UploadDockerImages.yml index 64c363b37..53e25bf65 100644 --- a/.github/workflows/UploadDockerImages.yml +++ b/.github/workflows/UploadDockerImages.yml @@ -40,16 +40,3 @@ jobs: - name: build maxdiffusion jax nightly image run: | bash .github/workflows/build_and_upload_images.sh CLOUD_IMAGE_NAME=maxdiffusion_jax_nightly MODE=nightly PROJECT=tpu-prod-env-multipod LOCAL_IMAGE_NAME=maxdiffusion_jax_nightly - - build-gpu-image: - runs-on: ["self-hosted", "e2", "cpu"] - steps: - - uses: actions/checkout@v3 - - name: Cleanup old docker images - run: docker system prune --all --force - - name: build maxdiffusion jax stable stack gpu image - run: | - bash .github/workflows/build_and_upload_images.sh CLOUD_IMAGE_NAME=maxdiffusion_jax_stable_gpu MODE=stable PROJECT=tpu-prod-env-multipod LOCAL_IMAGE_NAME=maxdiffusion_jax_stable_gpu DEVICE=gpu - - name: build maxdiffusion jax nightly image - run: | - bash .github/workflows/build_and_upload_images.sh CLOUD_IMAGE_NAME=maxdiffusion_jax_nightly_gpu MODE=nightly PROJECT=tpu-prod-env-multipod LOCAL_IMAGE_NAME=maxdiffusion_jax_nightly DEVICE=gpu diff --git a/README.md b/README.md index 15667de4e..4ad3c388f 100644 --- a/README.md +++ b/README.md @@ -52,6 +52,8 @@ MaxDiffusion supports * Wan2.1 text2vid (training and inference). * Wan2.2 text2vid (inference). +**Note on GPU Support:** GPU support is not actively maintained, but contributions are welcome + # Table of Contents