Add _repeated_blocks to Wan Animate transformer for regional compilation

dg845 · dg845 · commit bdbd14191ae8 · 2025-11-04T07:30:24.000+01:00
diff --git a/src/diffusers/models/transformers/transformer_wan_animate.py b/src/diffusers/models/transformers/transformer_wan_animate.py
@@ -572,6 +572,7 @@ class WanAnimateTransformer3DModel(
         "motion_synthesis_weight",
     ]
     _keys_to_ignore_on_load_unexpected = ["norm_added_q"]
+    _repeated_blocks = ["WanTransformerBlock"]
 
     @register_to_config
     def __init__(

Original file line number	Diff line number	Diff line change
`@@ -572,6 +572,7 @@ class WanAnimateTransformer3DModel(`
`572`	`572`	`"motion_synthesis_weight",`
`573`	`573`	`]`
`574`	`574`	`_keys_to_ignore_on_load_unexpected = ["norm_added_q"]`
	`575`	`+ _repeated_blocks = ["WanTransformerBlock"]`
`575`	`576`
`576`	`577`	`@register_to_config`
`577`	`578`	`def __init__(`