[bart-tiny-random] Put a 5MB model on S3 to allow faster exampl… (#3488)

2020-03-30 12:28:27 -04:00
parent 1f72865726
commit 8deff3acf2
3 changed files with 23 additions and 6 deletions
--- a/examples/summarization/bart/test_bart_examples.py
+++ b/examples/summarization/bart/test_bart_examples.py
@@ -25,7 +25,8 @@ class TestBartExamples(unittest.TestCase):
        tmp = Path(tempfile.gettempdir()) / "utest_generations_bart_sum.hypo"
        with tmp.open("w") as f:
            f.write("\n".join(articles))
-        testargs = ["evaluate_cnn.py", str(tmp), output_file_name]
+
+        testargs = ["evaluate_cnn.py", str(tmp), output_file_name, "sshleifer/bart-tiny-random"]
        with patch.object(sys, "argv", testargs):
            _run_generate()
            self.assertTrue(Path(output_file_name).exists())