Skip to content

Commit 6e3b58b

Browse files
Remove GPU image build steps from workflow and update README to clarify GPU support status (#335)
1 parent 3074581 commit 6e3b58b

2 files changed

Lines changed: 2 additions & 13 deletions

File tree

.github/workflows/UploadDockerImages.yml

Lines changed: 0 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -40,16 +40,3 @@ jobs:
4040
- name: build maxdiffusion jax nightly image
4141
run: |
4242
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
43-
44-
build-gpu-image:
45-
runs-on: ["self-hosted", "e2", "cpu"]
46-
steps:
47-
- uses: actions/checkout@v3
48-
- name: Cleanup old docker images
49-
run: docker system prune --all --force
50-
- name: build maxdiffusion jax stable stack gpu image
51-
run: |
52-
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
53-
- name: build maxdiffusion jax nightly image
54-
run: |
55-
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

README.md

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,8 @@ MaxDiffusion supports
5252
* Wan2.1 text2vid (training and inference).
5353
* Wan2.2 text2vid (inference).
5454

55+
**Note on GPU Support:** GPU support is not actively maintained, but contributions are welcome
56+
5557

5658
# Table of Contents
5759

0 commit comments

Comments
 (0)