diff --git a/Chrono/chronoedit_diffusers/transformer_chronoedit.py b/Chrono/chronoedit_diffusers/transformer_chronoedit.py index 07cd20b..efd93d8 100644 --- a/Chrono/chronoedit_diffusers/transformer_chronoedit.py +++ b/Chrono/chronoedit_diffusers/transformer_chronoedit.py @@ -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,