From 3845f886d8f45254ec4f61c46857d3acaa12dfbf Mon Sep 17 00:00:00 2001
From: smthemex <138738845+smthemex@users.noreply.github.com>
Date: Thu, 13 Nov 2025 19:12:19 +0800
Subject: [PATCH] Update transformer_chronoedit.py

---
 Chrono/chronoedit_diffusers/transformer_chronoedit.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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,