Fix slow tests v2 (#8746)

* Fix BART test * Fix MBART tests * Remove erroneous line from yaml * Update tests/test_modeling_bart.py * Quality
2020-11-24 09:35:12 -05:00
parent 2c83b3c38d
commit 6fdd0bb231
3 changed files with 4 additions and 3 deletions
--- a/tests/test_modeling_mbart.py
+++ b/tests/test_modeling_mbart.py
@@ -62,7 +62,7 @@ class AbstractSeq2SeqIntegrationTest(unittest.TestCase):

    @classmethod
    def setUpClass(cls):
-        cls.tokenizer = AutoTokenizer.from_pretrained(cls.checkpoint_name)
+        cls.tokenizer = AutoTokenizer.from_pretrained(cls.checkpoint_name, use_fast=False)
        return cls

    @cached_property