Skip to content

GitLab

Unverified Commit 9406c7bc authored by Sanchit Gandhi's avatar Sanchit Gandhi Committed by GitHub
Browse files

[modelcard] Update for ASR (#19985) 

* [modelcard] Update for ASR

* style
parent 225c36fb
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 0 deletions
src/transformers/modelcard.py
View file @ 9406c7bc
...@@ -32,6 +32,7 @@ from . import __version__ ...@@ -32,6 +32,7 @@ from . import __version__
from .models.auto.modeling_auto import ( from .models.auto.modeling_auto import (
MODEL_FOR_AUDIO_CLASSIFICATION_MAPPING_NAMES, MODEL_FOR_AUDIO_CLASSIFICATION_MAPPING_NAMES,
MODEL_FOR_CAUSAL_LM_MAPPING_NAMES, MODEL_FOR_CAUSAL_LM_MAPPING_NAMES,
MODEL_FOR_CTC_MAPPING_NAMES,
MODEL_FOR_IMAGE_CLASSIFICATION_MAPPING_NAMES, MODEL_FOR_IMAGE_CLASSIFICATION_MAPPING_NAMES,
MODEL_FOR_IMAGE_SEGMENTATION_MAPPING_NAMES, MODEL_FOR_IMAGE_SEGMENTATION_MAPPING_NAMES,
MODEL_FOR_MASKED_LM_MAPPING_NAMES, MODEL_FOR_MASKED_LM_MAPPING_NAMES,
...@@ -39,6 +40,7 @@ from .models.auto.modeling_auto import ( ...@@ -39,6 +40,7 @@ from .models.auto.modeling_auto import (
MODEL_FOR_QUESTION_ANSWERING_MAPPING_NAMES, MODEL_FOR_QUESTION_ANSWERING_MAPPING_NAMES,
MODEL_FOR_SEQ_TO_SEQ_CAUSAL_LM_MAPPING_NAMES, MODEL_FOR_SEQ_TO_SEQ_CAUSAL_LM_MAPPING_NAMES,
MODEL_FOR_SEQUENCE_CLASSIFICATION_MAPPING_NAMES, MODEL_FOR_SEQUENCE_CLASSIFICATION_MAPPING_NAMES,
MODEL_FOR_SPEECH_SEQ_2_SEQ_MAPPING_NAMES,
MODEL_FOR_TABLE_QUESTION_ANSWERING_MAPPING_NAMES, MODEL_FOR_TABLE_QUESTION_ANSWERING_MAPPING_NAMES,
MODEL_FOR_TOKEN_CLASSIFICATION_MAPPING_NAMES, MODEL_FOR_TOKEN_CLASSIFICATION_MAPPING_NAMES,
) )
...@@ -67,6 +69,7 @@ TASK_MAPPING = { ...@@ -67,6 +69,7 @@ TASK_MAPPING = {
"table-question-answering": MODEL_FOR_TABLE_QUESTION_ANSWERING_MAPPING_NAMES, "table-question-answering": MODEL_FOR_TABLE_QUESTION_ANSWERING_MAPPING_NAMES,
"token-classification": MODEL_FOR_TOKEN_CLASSIFICATION_MAPPING_NAMES, "token-classification": MODEL_FOR_TOKEN_CLASSIFICATION_MAPPING_NAMES,
"audio-classification": MODEL_FOR_AUDIO_CLASSIFICATION_MAPPING_NAMES, "audio-classification": MODEL_FOR_AUDIO_CLASSIFICATION_MAPPING_NAMES,
"automatic-speech-recognition": {**MODEL_FOR_CTC_MAPPING_NAMES, **MODEL_FOR_SPEECH_SEQ_2_SEQ_MAPPING_NAMES},
} }
logger = logging.get_logger(__name__) logger = logging.get_logger(__name__)
...@@ -273,6 +276,7 @@ TASK_TAG_TO_NAME_MAPPING = { ...@@ -273,6 +276,7 @@ TASK_TAG_TO_NAME_MAPPING = {
"token-classification": "Token Classification", "token-classification": "Token Classification",
"translation": "Translation", "translation": "Translation",
"zero-shot-classification": "Zero Shot Classification", "zero-shot-classification": "Zero Shot Classification",
"automatic-speech-recognition": "Automatic Speech Recognition",
} }
...@@ -287,6 +291,7 @@ METRIC_TAGS = [ ...@@ -287,6 +291,7 @@ METRIC_TAGS = [
"rouge", "rouge",
"sacrebleu", "sacrebleu",
"spearmanr", "spearmanr",
"wer",
] ]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment