SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Add AutoModelForZeroShotImageClassification (#22087)

Browse Source 
Adds AutoModelForZeroShotImageClassification to transformers
This commit is contained in:
Alara Dirik
2023-03-13 12:46:14 +03:00
committed by GitHub
parent b90fbc7e0b
commit 32e3466d38
12 changed files with 98 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
utils/update_metadata.py
View File

@@ -93,8 +93,8 @@ PIPELINE_TAGS_AND_AUTO_MODELS = [
("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",
"MODEL_FOR_ZERO_SHOT_IMAGE_CLASSIFICATION_MAPPING_NAMES",
"AutoModelForZeroShotImageClassification",
),
("depth-estimation", "MODEL_FOR_DEPTH_ESTIMATION_MAPPING_NAMES", "AutoModelForDepthEstimation"),
("video-classification", "MODEL_FOR_VIDEO_CLASSIFICATION_MAPPING_NAMES", "AutoModelForVideoClassification"),