Update src/transformers/__init__.py

Co-authored-by: NielsRogge <[email protected]>
NielsRogge · Sep 3, 2024 · 7ffa504 · 7ffa504
1 parent 533d298
commit 7ffa504
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/transformers/__init__.py b/src/transformers/__init__.py
@@ -775,8 +775,8 @@
     "models.vit_msn": ["ViTMSNConfig"],
     "models.vitdet": ["VitDetConfig"],
     "models.vitmatte": ["VitMatteConfig"],
-    "models.vitpose": ["ViTPoseConfig"],
-    "models.vitpose_backbone": ["ViTPoseBackboneConfig"],
+    "models.vitpose": ["VitPoseConfig"],
+    "models.vitpose_backbone": ["VitPoseBackboneConfig"],
     "models.vits": [
         "VitsConfig",
         "VitsTokenizer",