Automatically tag CLIP repos as zero-shot-image-classification (#19064)

* Add CLIP to zero-shot-image-classification * Make mapping private as it's not used for AutoClassing
2022-09-16 15:40:38 +02:00
parent 820cb97a3f
commit bc5d0b1046
2 changed files with 12 additions and 0 deletions
--- a/utils/update_metadata.py
+++ b/utils/update_metadata.py
@@ -91,6 +91,11 @@ PIPELINE_TAGS_AND_AUTO_MODELS = [
        "AutoModelForVisualQuestionAnswering",
    ),
    ("image-to-text", "MODEL_FOR_FOR_VISION_2_SEQ_MAPPING_NAMES", "AutoModelForVision2Seq"),
+    (
+        "zero-shot-image-classification",
+        "_MODEL_FOR_ZERO_SHOT_IMAGE_CLASSIFICATION_MAPPING_NAMES",
+        "AutoModel",
+    ),
 ]