mirrors/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-09-12 12:37:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

WIP Wan 2.2 S2V model. (#9568)

Browse Source
This commit is contained in:
comfyanonymous
2025-08-26 22:10:34 -07:00
committed by GitHub
parent 6a193ac557
commit 88aee596a3
5 changed files with 707 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/model_detection.py
View File

@@ -368,6 +368,8 @@ def detect_unet_config(state_dict, key_prefix, metadata=None):
dit_config["model_type"] = "camera"
else:
dit_config["model_type"] = "camera_2.2"
elif '{}casual_audio_encoder.encoder.final_linear.weight'.format(key_prefix) in state_dict_keys:
dit_config["model_type"] = "s2v"
else:
if '{}img_emb.proj.0.bias'.format(key_prefix) in state_dict_keys:
dit_config["model_type"] = "i2v"