adding multislice test with @parambole

GoogleCloudPlatform · Oct 2, 2024 · fbcface · fbcface
1 parent 746295e
commit fbcface
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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():