diff --git a/dags/imagegen_devx/maxdiffusion_e2e.py b/dags/imagegen_devx/maxdiffusion_e2e.py index bdbcae13..edb83aba 100644 --- a/dags/imagegen_devx/maxdiffusion_e2e.py +++ b/dags/imagegen_devx/maxdiffusion_e2e.py @@ -36,6 +36,7 @@ maxdiffusion_test_configs = { # accelerator: list of slices to test "v6e-256": [1], + "v4-8": [2], } current_datetime = config.get_current_datetime() for accelerator, slices in maxdiffusion_test_configs.items():