Unverified Commit d5888ef0 authored by Nicolas Patry's avatar Nicolas Patry Committed by GitHub
Browse files

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

`encoder_no_repeat_ngram_size` from their config.
parent 8c3b1fcb
......@@ -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 (
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment