We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
TASK_MAPPING
1 parent 41f5c32 commit ec42fffCopy full SHA for ec42fff
src/transformers/modelcard.py
@@ -33,6 +33,7 @@
33
MODEL_FOR_CTC_MAPPING_NAMES,
34
MODEL_FOR_IMAGE_CLASSIFICATION_MAPPING_NAMES,
35
MODEL_FOR_IMAGE_SEGMENTATION_MAPPING_NAMES,
36
+ MODEL_FOR_IMAGE_TEXT_TO_TEXT_MAPPING_NAMES,
37
MODEL_FOR_MASKED_LM_MAPPING_NAMES,
38
MODEL_FOR_OBJECT_DETECTION_MAPPING_NAMES,
39
MODEL_FOR_QUESTION_ANSWERING_MAPPING_NAMES,
@@ -70,6 +71,7 @@
70
71
"audio-classification": MODEL_FOR_AUDIO_CLASSIFICATION_MAPPING_NAMES,
72
"automatic-speech-recognition": {**MODEL_FOR_CTC_MAPPING_NAMES, **MODEL_FOR_SPEECH_SEQ_2_SEQ_MAPPING_NAMES},
73
"zero-shot-image-classification": MODEL_FOR_ZERO_SHOT_IMAGE_CLASSIFICATION_MAPPING_NAMES,
74
+ "image-text-to-text": MODEL_FOR_IMAGE_TEXT_TO_TEXT_MAPPING_NAMES,
75
}
76
77
logger = logging.get_logger(__name__)
0 commit comments