Skip test while the model is not available (#12740)

2021-07-15 15:14:12 +02:00
parent 8c7bd1b97b
commit eb2e006b35
1 changed files with 1 additions and 0 deletions
--- a/tests/test_modeling_megatron_gpt2.py
+++ b/tests/test_modeling_megatron_gpt2.py
@@ -31,6 +31,7 @@ if is_torch_available():
@require_tokenizers
 class MegatronGPT2IntegrationTest(unittest.TestCase):
    @slow
+    @unittest.skip("Model is not available.")
    def test_inference_no_head(self):
        directory = "nvidia/megatron-gpt2-345m/"
        if "MYDIR" in os.environ: