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

Patch models (#6326)

Browse Source 
* TFAlbertFor{TokenClassification, MultipleChoice}

* Patch models

* BERT and TF BERT info


s

* Update check_repo
This commit is contained in:
Lysandre Debut
2020-08-10 10:39:17 -04:00
committed by GitHub
parent 6028ed92bd
commit b99098abc7
6 changed files with 37 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
utils/check_repo.py
View File

@@ -18,12 +18,6 @@ IGNORE_NON_TESTED = [
"DPRSpanPredictor", # Building part of bigger (tested) model.
"ReformerForMaskedLM", # Needs to be setup as decoder.
"T5Stack", # Building part of bigger (tested) model.
"TFAlbertForMultipleChoice", # TODO: fix
"TFAlbertForTokenClassification", # TODO: fix
"TFBertLMHeadModel", # TODO: fix
"TFElectraForMultipleChoice", # Fix is in #6284
"TFElectraForQuestionAnswering", # TODO: fix
"TFElectraForSequenceClassification", # Fix is in #6284
"TFElectraMainLayer", # Building part of bigger (tested) model (should it be a TFPreTrainedModel ?)
"TFRobertaForMultipleChoice", # TODO: fix
]