riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 33 Packages Projects Releases Wiki Activity

Hotfixing tests (blenderbot decoderonly tests, also need to remove (#10003)

Browse Source 
`encoder_no_repeat_ngram_size` from their config.
This commit is contained in:
Nicolas Patry 2021-02-04 17:41:34 +01:00 committed by GitHub
parent 8c3b1fcb67
commit d5888ef0ab
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/test_modeling_blenderbot.py
View File

@ -325,6 +325,7 @@ class BlenderbotStandaloneDecoderModelTester:
decoder_attention_heads=4,
max_position_embeddings=30,
is_encoder_decoder=False,
encoder_no_repeat_ngram_size=0,
pad_token_id=0,
bos_token_id=1,
eos_token_id=2,
@ -355,6 +356,7 @@ class BlenderbotStandaloneDecoderModelTester:
self.use_cache = use_cache
self.max_position_embeddings = max_position_embeddings
self.is_encoder_decoder = is_encoder_decoder
self.encoder_no_repeat_ngram_size = encoder_no_repeat_ngram_size
self.scope = None
self.decoder_key_length = decoder_seq_length
@ -386,6 +388,7 @@ class BlenderbotStandaloneDecoderModelTester:
decoder_start_token_id=self.decoder_start_token_id,
max_position_embeddings=self.max_position_embeddings,
is_encoder_decoder=self.is_encoder_decoder,
encoder_no_repeat_ngram_size=self.encoder_no_repeat_ngram_size,
)
return (