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

[Pretrained Model] Add resize_position_embeddings (#13559)

Browse Source 
* finish

* delete bogus file

* correct some stuff

* finish

* finish
This commit is contained in:
Patrick von Platen
2021-09-15 19:03:56 +02:00
committed by GitHub
parent c783e14887
commit 95f933ea85
8 changed files with 413 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_modeling_pegasus.py
View File

@@ -229,6 +229,7 @@ class PegasusModelTest(ModelTesterMixin, GenerationTesterMixin, unittest.TestCas
all_model_classes = (PegasusModel, PegasusForConditionalGeneration) if is_torch_available() else ()
all_generative_model_classes = (PegasusForConditionalGeneration,) if is_torch_available() else ()
is_encoder_decoder = True
test_resize_position_embeddings = True
test_pruning = False
test_missing_keys = False
@@ -526,6 +527,7 @@ class PegasusStandaloneDecoderModelTester:
class PegasusStandaloneDecoderModelTest(ModelTesterMixin, GenerationTesterMixin, unittest.TestCase):
all_model_classes = (PegasusDecoder, PegasusForCausalLM) if is_torch_available() else ()
all_generative_model_classes = (PegasusForCausalLM,) if is_torch_available() else ()
test_resize_position_embeddings = True
test_pruning = False
is_encoder_decoder = False