Reverted the bigbird comment

huggingface · Dec 20, 2022 · 34d12bf · 34d12bf
1 parent 5bf069f
commit 34d12bf
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/tests/exporters/test_onnx_export.py b/tests/exporters/test_onnx_export.py
@@ -107,7 +107,7 @@
     # Not using facebook/blenderbot-400M-distill because it takes too much time for testing, and might cause OOM.
     ("blenderbot", "facebook/blenderbot-90M"),
     # Not using google/bigbird-pegasus-large-arxiv because it takes too much time for testing.
-    # ("bigbird-pegasus", "hf-internal-testing/tiny-random-bigbird_pegasus"),
+    ("bigbird-pegasus", "hf-internal-testing/tiny-random-bigbird_pegasus"),
     # Not using google/long-t5-local-base because it takes too much time for testing.
     ("longt5", "hf-internal-testing/tiny-random-LongT5Model"),
     ("whisper", "openai/whisper-tiny.en"),
@@ -137,11 +137,11 @@
         "hf-internal-testing/tiny-random-m2m_100",
     ),
     # Not using google/bigbird-pegasus-large-arxiv because it takes too much time for testing.
-    # (
-    #     "bigbird-pegasus",
-    #     "hf-internal-testing/tiny-random-bigbird_pegasus",
-    #     ("seq2seq-lm", "seq2seq-lm-with-past"),
-    # ),
+    (
+        "bigbird-pegasus",
+        "hf-internal-testing/tiny-random-bigbird_pegasus",
+        ("seq2seq-lm", "seq2seq-lm-with-past"),
+    ),
     (
         "pegasus",
         "hf-internal-testing/tiny-random-PegasusModel",