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

Make ProphetNetModel really compatible with EncoderDecoder (#9033)

Browse Source 
* improve

* finish

* upload model

* fix lm head

* fix test
This commit is contained in:
Patrick von Platen
2020-12-11 16:59:54 +01:00
committed by GitHub
parent 24f6cdeab6
commit 9cc9f4122e
5 changed files with 59 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
utils/check_repo.py
View File

@@ -34,6 +34,7 @@ IGNORE_NON_TESTED = [
"BertLMHeadModel", # Needs to be setup as decoder.
"DPREncoder", # Building part of bigger (tested) model.
"DPRSpanPredictor", # Building part of bigger (tested) model.
"ProphetNetDecoderWrapper", # Building part of bigger (tested) model.
"ReformerForMaskedLM", # Needs to be setup as decoder.
"T5Stack", # Building part of bigger (tested) model.
"TFDPREncoder", # Building part of bigger (tested) model.
@@ -74,6 +75,7 @@ IGNORE_NON_AUTO_CONFIGURED = [
"OpenAIGPTDoubleHeadsModel",
"ProphetNetDecoder",
"ProphetNetEncoder",
"ProphetNetDecoderWrapper",
"RagModel",
"RagSequenceForGeneration",
"RagTokenForGeneration",