revert

Author: prishajain1

src/maxdiffusion/pipelines/wan/wan_pipeline_2_2.py

@@ -41,15 +41,15 @@ def _load_and_init(cls, config, restored_checkpoint=None, vae_only=False, load_t
             rngs=common_components["rngs"],
             config=config,
             restored_checkpoint=restored_checkpoint,
-            subfolder="transformer_2"
+            subfolder="transformer"
         )
         high_noise_transformer = super().load_transformer(
             devices_array=common_components["devices_array"],
             mesh=common_components["mesh"],
             rngs=common_components["rngs"],
             config=config,
             restored_checkpoint=restored_checkpoint,
-            subfolder="transformer"
+            subfolder="transformer_2"
         )
 
         pipeline = cls(