From 725e6ec60621c6f001af04769173e7dbb3c53541 Mon Sep 17 00:00:00 2001
From: comfyanonymous <121283862+comfyanonymous@users.noreply.github.com>
Date: Wed, 12 Aug 2026 13:22:50 -0700
Subject: [PATCH] Support  anima tunes with extra blocks. (#15555)

---
 comfy/model_detection.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/comfy/model_detection.py b/comfy/model_detection.py
index bc7b2b9f8..aec205290 100644
--- a/comfy/model_detection.py
+++ b/comfy/model_detection.py
@@ -830,11 +830,10 @@ def detect_unet_config(state_dict, key_prefix, metadata=None):
 
         dit_config["use_adaln_lora"] = True
         dit_config["adaln_lora_dim"] = 256
+        dit_config["num_blocks"] = count_blocks(state_dict_keys, '{}blocks.'.format(key_prefix) + '{}.')
         if dit_config["model_channels"] == 2048:
-            dit_config["num_blocks"] = 28
             dit_config["num_heads"] = 16
         elif dit_config["model_channels"] == 5120:
-            dit_config["num_blocks"] = 36
             dit_config["num_heads"] = 40
 
         if dit_config["in_channels"] == 16: