File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 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
Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments