skip gptbigcode in test

huggingface · Dec 22, 2023 · e128c16 · e128c16
1 parent d32805e
commit e128c16
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/tests/generation/test_utils.py b/tests/generation/test_utils.py
@@ -1505,11 +1505,9 @@ def test_beam_search_low_memory(self):
             if any(
                 model_name in model_class.__name__.lower()
                 for model_name in [
-                    "fsmt",
-                    "reformer",
                     "bloom",
                     "ctrl",
-                    "gpt_bigcode",
+                    "gptbigcode",
                     "transo_xl",
                     "xlnet",
                     "cpm",