Update transformer_chronoedit.py

2025-11-13 19:12:19 +08:00
parent 12cb2c85f9
commit 3845f886d8
1 changed files with 1 additions and 1 deletions
@@ -338,7 +338,7 @@ class WanTransformer3DModel(ModelMixin, ConfigMixin, PeftAdapterMixin, FromOrigi
    _no_split_modules = ["ChronoEditTransformerBlock"]
    _keep_in_fp32_modules = ["time_embedder", "scale_shift_table", "norm1", "norm2", "norm3"]
    _keys_to_ignore_on_load_unexpected = ["norm_added_q"]
-
+    _repeated_blocks = ["WanTransformerBlock"]
    @register_to_config
    def __init__(
        self,