Refactor: Rename MagiTransformer classes to Magi1Transformer for consistency

Author: tolgacangoz

tests/models/transformers/test_models_transformer_magi1.py

@@ -28,7 +28,7 @@
 enable_full_determinism()
 
 
-class MagiTransformer3DTests(ModelTesterMixin, unittest.TestCase):
+class Magi1Transformer3DTests(ModelTesterMixin, unittest.TestCase):
     model_class = Magi1Transformer3DModel
     main_input_name = "hidden_states"
     uses_custom_attn_processor = True
@@ -84,8 +84,8 @@ def test_gradient_checkpointing_is_applied(self):
         super().test_gradient_checkpointing_is_applied(expected_set=expected_set)
 
 
-class MagiTransformerCompileTests(TorchCompileTesterMixin, unittest.TestCase):
+class Magi1TransformerCompileTests(TorchCompileTesterMixin, unittest.TestCase):
     model_class = Magi1Transformer3DModel
 
     def prepare_init_args_and_inputs_for_common(self):
-        return MagiTransformer3DTests().prepare_init_args_and_inputs_for_common()
+        return Magi1Transformer3DTests().prepare_init_args_and_inputs_for_common()

tests/single_file/test_model_magi_transformer3d_single_file.py

@@ -25,7 +25,7 @@
 )
 
 
-class MagiTransformer3DModelText2VideoSingleFileTest(unittest.TestCase):
+class Magi1Transformer3DModelText2VideoSingleFileTest(unittest.TestCase):
     model_class = Magi1Transformer3DModel
     ckpt_path = "https://huggingface.co/sand-ai/MAGI-1/blob/main/transformer/diffusion_pytorch_model.safetensors"
     repo_id = "sand-ai/MAGI-1"