PR #2183 ViTPose - SemanticDiff

ViTPose #2183

echarlaix merged 8 commits into huggingface:main from ariG23498:aritra/vitpose-onnx

Add ONNX export support for ViTPose

c0df0450

building dummy inputs for vit post

cccc05ce

ariG23498 requested a review from

xenova 322 days ago

xenova commented on 2025-02-10

ariG23498 changed the base branch from vitpose-onnx to main 322 days ago

resolve merge conflicts

23e3b12b

ariG23498 requested a review from

xenova 322 days ago

Move vitpose config to custom class

10e9ba67

Move input generators

13b787e4

Patch VitPose models with num_experts>1

2f7a0a4d

Formatting

130dfaed

Add vitpose export unit tests

30e163a7

xenova approved these changes on 2025-02-11

xenova requested a review from

echarlaix 321 days ago

echarlaix approved these changes on 2025-02-18

echarlaix merged 02456194 into main 314 days ago

Reviewers

echarlaix

xenova

Assignees

No one assigned

Labels

None yet

Milestone

No milestone