optimum
02456194 - Add ViTPose ONNX export (#2183)

Commit
314 days ago
Add ViTPose ONNX export (#2183) * Add ONNX export support for ViTPose * building dummy inputs for vit post * Move vitpose config to custom class * Move input generators * Patch VitPose models with num_experts>1 * Formatting * Add vitpose export unit tests --------- Co-authored-by: Joshua Lochner <admin@xenova.com>
Author
Parents
Loading