From 8c7bd1b97b294dba9743894166bacb60f247fec9 Mon Sep 17 00:00:00 2001
From: Lysandre Debut <lysandre@huggingface.co>
Date: Thu, 15 Jul 2021 15:06:47 +0200
Subject: [PATCH] Skip test while the model is not available (#12739)

---
 tests/test_modeling_megatron_bert.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tests/test_modeling_megatron_bert.py b/tests/test_modeling_megatron_bert.py
index 7a58e9f753..0cfc444e04 100644
--- a/tests/test_modeling_megatron_bert.py
+++ b/tests/test_modeling_megatron_bert.py
@@ -356,6 +356,7 @@ TOLERANCE = 1e-4
 @require_tokenizers
 class MegatronBertModelIntegrationTests(unittest.TestCase):
     @slow
+    @unittest.skip("Model is not available.")
     def test_inference_no_head(self):
         directory = "nvidia/megatron-bert-uncased-345m"
         if "MYDIR" in os.environ: