Fix: ensure consistent dtype and eval mode in pipeline save/load tests (#13339)

* Fix: ensure consistent dtype and eval mode in pipeline save/load tests

* Modify according to the comments

* Update according to the comments

* Update comment

* Code quality

* cast buffers to torch.float16

* conflict

* Fix

---------

Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>

Author: YangKai0616

src/diffusers/models/transformers/transformer_wan_animate.py

@@ -1029,6 +1029,7 @@ class WanAnimateTransformer3DModel(
         "norm2",
         "norm3",
         "motion_synthesis_weight",
+        "rope",
     ]
     _keys_to_ignore_on_load_unexpected = ["norm_added_q"]
     _repeated_blocks = ["WanTransformerBlock"]

tests/pipelines/test_pipelines_common.py

@@ -1443,10 +1443,24 @@ def test_save_load_float16(self, expected_max_diff=1e-2):
                         param.data = param.data.to(torch_device).to(torch.float32)
                     else:
                         param.data = param.data.to(torch_device).to(torch.float16)
+                for name, buf in module.named_buffers():
+                    if not buf.is_floating_point():
+                        buf.data = buf.data.to(torch_device)
+                    elif any(
+                        module_to_keep_in_fp32 in name.split(".")
+                        for module_to_keep_in_fp32 in module._keep_in_fp32_modules
+                    ):
+                        buf.data = buf.data.to(torch_device).to(torch.float32)
+                    else:
+                        buf.data = buf.data.to(torch_device).to(torch.float16)
 
             elif hasattr(module, "half"):
                 components[name] = module.to(torch_device).half()
 
+        for key, component in components.items():
+            if hasattr(component, "eval"):
+                component.eval()
+
         pipe = self.pipeline_class(**components)
         for component in pipe.components.values():
             if hasattr(component, "set_default_attn_processor"):