Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
transformers
Commits
9406c7bc
Unverified
Commit
9406c7bc
authored
Oct 31, 2022
by
Sanchit Gandhi
Committed by
GitHub
Oct 31, 2022
Browse files
[modelcard] Update for ASR (#19985)
* [modelcard] Update for ASR * style
parent
225c36fb
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
5 additions
and
0 deletions
+5
-0
src/transformers/modelcard.py
src/transformers/modelcard.py
+5
-0
No files found.
src/transformers/modelcard.py
View file @
9406c7bc
...
...
@@ -32,6 +32,7 @@ from . import __version__
from
.models.auto.modeling_auto
import
(
MODEL_FOR_AUDIO_CLASSIFICATION_MAPPING_NAMES
,
MODEL_FOR_CAUSAL_LM_MAPPING_NAMES
,
MODEL_FOR_CTC_MAPPING_NAMES
,
MODEL_FOR_IMAGE_CLASSIFICATION_MAPPING_NAMES
,
MODEL_FOR_IMAGE_SEGMENTATION_MAPPING_NAMES
,
MODEL_FOR_MASKED_LM_MAPPING_NAMES
,
...
...
@@ -39,6 +40,7 @@ from .models.auto.modeling_auto import (
MODEL_FOR_QUESTION_ANSWERING_MAPPING_NAMES
,
MODEL_FOR_SEQ_TO_SEQ_CAUSAL_LM_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_TOKEN_CLASSIFICATION_MAPPING_NAMES
,
)
...
...
@@ -67,6 +69,7 @@ TASK_MAPPING = {
"table-question-answering"
:
MODEL_FOR_TABLE_QUESTION_ANSWERING_MAPPING_NAMES
,
"token-classification"
:
MODEL_FOR_TOKEN_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__
)
...
...
@@ -273,6 +276,7 @@ TASK_TAG_TO_NAME_MAPPING = {
"token-classification"
:
"Token Classification"
,
"translation"
:
"Translation"
,
"zero-shot-classification"
:
"Zero Shot Classification"
,
"automatic-speech-recognition"
:
"Automatic Speech Recognition"
,
}
...
...
@@ -287,6 +291,7 @@ METRIC_TAGS = [
"rouge"
,
"sacrebleu"
,
"spearmanr"
,
"wer"
,
]
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment