From af0afcb48bc04eb3df3c2a8170f31301cbd222f3 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Quentin=20Gallou=C3=A9dec?= <45557362+qgallouedec@users.noreply.github.com> Date: Fri, 26 Jul 2024 11:33:07 +0200 Subject: [PATCH] Skip BigBird save and load test until next transformers version (#1874) --- tests/test_modeling_value_head.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_modeling_value_head.py b/tests/test_modeling_value_head.py index d7c2b5cf6f6..6f580e5a7fd 100644 --- 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",