Unverified Commit ab5d06a0 authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

[T5, examples] replace heavy t5 models with tiny random models (#3556)

* replace heavy t5 models with tiny random models as was done by sshleifer

* fix isort
parent a4ee4da1
......@@ -29,7 +29,14 @@ class TestT5Examples(unittest.TestCase):
output_file_name = Path(tempfile.gettempdir()) / "utest_output_t5_sum.hypo"
score_file_name = Path(tempfile.gettempdir()) / "utest_score_t5_sum.hypo"
testargs = ["evaluate_cnn.py", "t5-small", str(tmp), str(output_file_name), str(tmp), str(score_file_name)]
testargs = [
"evaluate_cnn.py",
"patrickvonplaten/t5-tiny-random",
str(tmp),
str(output_file_name),
str(tmp),
str(score_file_name),
]
with patch.object(sys, "argv", testargs):
run_generate()
......
......@@ -37,7 +37,7 @@ class TestT5Examples(unittest.TestCase):
testargs = [
"evaluate_wmt.py",
"t5-small",
"patrickvonplaten/t5-tiny-random",
str(tmp_source),
str(output_file_name),
str(tmp_target),
......
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