Skip to content

Repository files navigation

ComfyUI helper nodes for Wan video 2.2 Animate preprocessing

Nodes to run the ViTPose model, get face crops and keypoint list for SAM2 segmentation.

Models:

to ComfyUI/models/detection (subject to change in the future)

YOLO:

https://huggingface.co/Wan-AI/Wan2.2-Animate-14B/blob/main/process_checkpoint/det/yolov10m.onnx

ViTPose ONNX:

Use either the Large model from here:

https://huggingface.co/JunkyByte/easy_ViTPose/tree/main/onnx/wholebody

Or the Huge model like in the original code, it's split into two files due to ONNX file size limit:

Both files need to be in same directory, and the onnx file selected in the model loader:

vitpose_h_wholebody_data.bin and vitpose_h_wholebody_model.onnx

https://huggingface.co/Kijai/vitpose_comfy/tree/main/onnx

example

About

ComfyUI nodes for WanAnimate model input preprocessing

Resources

Stars

544 stars

Watchers

6 watching

Forks

Releases

Packages

Contributors

Languages