From d5888ef0ab949ec82ed4768556c2b2743e3ca1df Mon Sep 17 00:00:00 2001
From: Nicolas Patry <patry.nicolas@protonmail.com>
Date: Thu, 4 Feb 2021 17:41:34 +0100
Subject: [PATCH] Hotfixing tests (blenderbot decoderonly tests, also need to
 remove (#10003)

`encoder_no_repeat_ngram_size` from their config.
---
 tests/test_modeling_blenderbot.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/tests/test_modeling_blenderbot.py b/tests/test_modeling_blenderbot.py
index 1e5337d65d..bff8f1ee0e 100644
--- a/tests/test_modeling_blenderbot.py
+++ b/tests/test_modeling_blenderbot.py
@@ -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 (