Skip to content
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.

Commit ec42fff

Browse files
saattrupdanstevhliu
authored andcommittedApr 1, 2025··
fix: Add 'image-text-to-text' to TASK_MAPPING
1 parent 41f5c32 commit ec42fff

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed
 

Diff for: ‎src/transformers/modelcard.py

+2
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,7 @@
3333
MODEL_FOR_CTC_MAPPING_NAMES,
3434
MODEL_FOR_IMAGE_CLASSIFICATION_MAPPING_NAMES,
3535
MODEL_FOR_IMAGE_SEGMENTATION_MAPPING_NAMES,
36+
MODEL_FOR_IMAGE_TEXT_TO_TEXT_MAPPING_NAMES,
3637
MODEL_FOR_MASKED_LM_MAPPING_NAMES,
3738
MODEL_FOR_OBJECT_DETECTION_MAPPING_NAMES,
3839
MODEL_FOR_QUESTION_ANSWERING_MAPPING_NAMES,
@@ -70,6 +71,7 @@
7071
"audio-classification": MODEL_FOR_AUDIO_CLASSIFICATION_MAPPING_NAMES,
7172
"automatic-speech-recognition": {**MODEL_FOR_CTC_MAPPING_NAMES, **MODEL_FOR_SPEECH_SEQ_2_SEQ_MAPPING_NAMES},
7273
"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,
7375
}
7476

7577
logger = logging.get_logger(__name__)

0 commit comments

Comments
 (0)
Please sign in to comment.