Basic WIP support for the wan animate model. (#9939)

This commit is contained in:
comfyanonymous
2025-09-19 00:07:17 -07:00
committed by GitHub
parent 711bcf33ee
commit dc95b6acc0
5 changed files with 666 additions and 1 deletions

View File

@@ -404,6 +404,8 @@ def detect_unet_config(state_dict, key_prefix, metadata=None):
dit_config["model_type"] = "s2v"
elif '{}audio_proj.audio_proj_glob_1.layer.bias'.format(key_prefix) in state_dict_keys:
dit_config["model_type"] = "humo"
elif '{}face_adapter.fuser_blocks.0.k_norm.weight'.format(key_prefix) in state_dict_keys:
dit_config["model_type"] = "animate"
else:
if '{}img_emb.proj.0.bias'.format(key_prefix) in state_dict_keys:
dit_config["model_type"] = "i2v"