Re-add BigBird Pegasus save/load test (huggingface#1882)

Co-authored-by: Quentin Gallouédec <[email protected]>
claralp · Jul 28, 2024 · b249c79 · b249c79
1 parent 004f78a
commit b249c79
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_modeling_value_head.py b/tests/test_modeling_value_head.py
@@ -38,7 +38,7 @@
 
 ALL_SEQ2SEQ_MODELS = [
     "trl-internal-testing/tiny-random-BartForConditionalGeneration",
-    # "trl-internal-testing/tiny-random-BigBirdPegasusForConditionalGeneration",
+    "trl-internal-testing/tiny-random-BigBirdPegasusForConditionalGeneration",
     "trl-internal-testing/tiny-random-BlenderbotForConditionalGeneration",
     "trl-internal-testing/tiny-random-BlenderbotSmallForConditionalGeneration",
     "trl-internal-testing/tiny-random-FSMTForConditionalGeneration",