Update tests/models/transformers/test_models_transformer_z_image.py

Co-authored-by: dg845 <58458699+dg845@users.noreply.github.com>

Author: sayakpaul

tests/models/transformers/test_models_transformer_z_image.py

@@ -140,7 +140,7 @@ def test_group_offloading_with_disk(self):
         super().test_group_offloading_with_disk()
 
 
-class Flux2TransformerCompileTests(TorchCompileTesterMixin, unittest.TestCase):
+class ZImageTransformerCompileTests(TorchCompileTesterMixin, unittest.TestCase):
     model_class = ZImageTransformer2DModel
     different_shapes_for_compilation = [(4, 4), (4, 8), (8, 8)]