transformer corrected in wan 2.2 t2v and config files updated

Author: prishajain1

src/maxdiffusion/configs/base_wan_14b.yml

@@ -29,6 +29,7 @@ log_period: 100
 
 pretrained_model_name_or_path: 'Wan-AI/Wan2.1-T2V-14B-Diffusers'
 model_name: wan2.1
+model_type: 'T2V'
 
 # Overrides the transformer from pretrained_model_name_or_path
 wan_transformer_pretrained_model_name_or_path: ''

src/maxdiffusion/configs/base_wan_27b.yml

@@ -29,6 +29,7 @@ log_period: 100
 
 pretrained_model_name_or_path: 'Wan-AI/Wan2.2-T2V-A14B-Diffusers'
 model_name: wan2.2
+model_type: 'T2V'
 
 # Overrides the transformer from pretrained_model_name_or_path
 wan_transformer_pretrained_model_name_or_path: ''

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"
+            subfolder="transformer_2"
         )
         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_2"
+            subfolder="transformer"
         )
 
         pipeline = cls(