zlei6/ComfyUI
1
0
Fork 0
mirror of https://github.com/Comfy-Org/ComfyUI.git synced 2026-08-18 23:38:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

Support anima tunes with extra blocks. (#15555)

Browse Source
This commit is contained in:
comfyanonymous
2026-08-12 13:22:50 -07:00
committed by GitHub
parent bd34f338ac
commit 725e6ec606
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
comfy/model_detection.py
View File

@@ -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: