Fixing the pipeline tests.

Narsil · Oct 4, 2021 · 605725f · 605725f
1 parent 12a1dd9
commit 605725f
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/tests/test_pipelines_common.py b/tests/test_pipelines_common.py
@@ -109,6 +109,11 @@ def gen_test(ModelClass, checkpoint, tiny_config, tokenizer_class, feature_extra
             def test(self):
                 if ModelClass.__name__.endswith("ForCausalLM"):
                     tiny_config.is_encoder_decoder = False
+                    if hasattr(tiny_config, "encoder_no_repeat_ngram_size"):
+                        # specific for blenderbot which supports both decoder-only
+                        # encoder/decoder but the test config  only reflects
+                        # encoder/decoder arch
+                        tiny_config.encoder_no_repeat_ngram_size = 0
                 if ModelClass.__name__.endswith("WithLMHead"):
                     tiny_config.is_decoder = True
                 model = ModelClass(tiny_config)