comfyanonymous/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-26 11:09:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Support the Anima model. (#12012)

Browse Source
This commit is contained in:
comfyanonymous
2026-01-21 16:44:28 -08:00
committed by GitHub
parent bdeac8897e
commit abe2ec26a6
6 changed files with 326 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/model_detection.py
View File

@@ -550,6 +550,8 @@ def detect_unet_config(state_dict, key_prefix, metadata=None):
if '{}blocks.0.mlp.layer1.weight'.format(key_prefix) in state_dict_keys: # Cosmos predict2
dit_config = {}
dit_config["image_model"] = "cosmos_predict2"
if "{}llm_adapter.blocks.0.cross_attn.q_proj.weight".format(key_prefix) in state_dict_keys:
dit_config["image_model"] = "anima"
dit_config["max_img_h"] = 240
dit_config["max_img_w"] = 240
dit_config["max_frames"] = 128