Add vitpose export unit tests

huggingface · Feb 11, 2025 · 30e163a · 30e163a
1 parent 130dfae
commit 30e163a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/exporters/exporters_utils.py b/tests/exporters/exporters_utils.py
@@ -173,6 +173,7 @@
     "vit-mae": "hf-internal-testing/tiny-random-ViTMAEModel",
     "vit-msn": "hf-internal-testing/tiny-random-ViTMSNForImageClassification",
     "vits": "echarlaix/tiny-random-vits",
+    "vitpose": "hf-internal-testing/tiny-random-VitPoseForPoseEstimation",
     "yolos": "hf-internal-testing/tiny-random-YolosModel",
     "whisper": "optimum-internal-testing/tiny-random-whisper",
     "hubert": "hf-internal-testing/tiny-random-HubertModel",
@@ -297,6 +298,7 @@
     "vit": "google/vit-base-patch16-224",
     "vit-mae": "facebook/vit-mae-base",
     "vit-msn": "facebook/vit-msn-small",
+    "vitpose": "usyd-community/vitpose-plus-small",
     "yolos": "hustvl/yolos-tiny",
     "whisper": "openai/whisper-tiny.en",
     "hubert": "facebook/hubert-base-ls960",