Skip to content

Commit fb46ce6

Browse files
remove MAXDIFFUSION_GPU_JAX_STABLE image and related DAG for GPU tests (#1195)
1 parent 26acd7b commit fb46ce6

File tree

2 files changed

+0
-80
lines changed

2 files changed

+0
-80
lines changed

dags/common/vm_resource.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -367,9 +367,6 @@ class DockerImage(enum.Enum):
367367
"gcr.io/tpu-prod-env-multipod/maxdiffusion_jax_stable_stack:"
368368
f"{datetime.datetime.today().strftime('%Y-%m-%d')}"
369369
)
370-
MAXDIFFUSION_GPU_JAX_STABLE = (
371-
"gcr.io/tpu-prod-env-multipod/maxdiffusion_jax_stable_gpu:latest"
372-
)
373370
MAXDIFFUSION_TPU_STABLE_STACK_NIGHTLY_JAX = (
374371
"gcr.io/tpu-prod-env-multipod/maxdiffusion_jax_stable_stack_nightly:"
375372
f"{datetime.datetime.today().strftime('%Y-%m-%d')}"

dags/sparsity_diffusion_devx/maxdiffusion_gpu_e2e.py

Lines changed: 0 additions & 77 deletions
This file was deleted.

0 commit comments

Comments
 (0)