@@ -1029,6 +1029,7 @@ class WanAnimateTransformer3DModel(
         "norm2",
         "norm3",
         "motion_synthesis_weight",
+        "rope",
     ]
     _keys_to_ignore_on_load_unexpected = ["norm_added_q"]
     _repeated_blocks = ["WanTransformerBlock"]
 
@@ -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"):
Original file line number	Diff line number	Diff line change
`@@ -1029,6 +1029,7 @@ class WanAnimateTransformer3DModel(`
`1029`	`1029`	`"norm2",`
`1030`	`1030`	`"norm3",`
`1031`	`1031`	`"motion_synthesis_weight",`
	`1032`	`+ "rope",`
`1032`	`1033`	`]`
`1033`	`1034`	`_keys_to_ignore_on_load_unexpected = ["norm_added_q"]`
`1034`	`1035`	`_repeated_blocks = ["WanTransformerBlock"]`