From 9e19885a6f85e050caf8d7e87af6587e7130ebe0 Mon Sep 17 00:00:00 2001
From: Haoming <hmstudy02@gmail.com>
Date: Sun, 23 Nov 2025 15:10:18 +0800
Subject: [PATCH] animate

---
 comfy/ldm/wan/model_animate.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/comfy/ldm/wan/model_animate.py b/comfy/ldm/wan/model_animate.py
index 7c87835d4..84d7adec4 100644
--- a/comfy/ldm/wan/model_animate.py
+++ b/comfy/ldm/wan/model_animate.py
@@ -523,7 +523,10 @@ class AnimateWanModel(WanModel):
 
         patches_replace = transformer_options.get("patches_replace", {})
         blocks_replace = patches_replace.get("dit", {})
+        transformer_options["total_blocks"] = len(self.blocks)
+        transformer_options["block_type"] = "double"
         for i, block in enumerate(self.blocks):
+            transformer_options["block_index"] = i
             if ("double_block", i) in blocks_replace:
                 def block_wrap(args):
                     out = {}