diff --git a/.github/workflows/UploadDockerImages.yml b/.github/workflows/UploadDockerImages.yml index 64c363b3..53e25bf6 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 15667de4..4ad3c388 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