RomanNum3ral
/
ComfyUI
mirror of https://github.com/Comfy-Org/ComfyUI.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Support anima tunes with extra blocks. (#15555)

This commit is contained in:
comfyanonymous 2026-08-12 13:22:50 -07:00 committed by GitHub
parent bd34f338ac
commit 725e6ec606
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 2 deletions
Show Stats 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: