Skip to content

GitLab

Unverified Commit 8c7bd1b9 authored by Lysandre Debut's avatar Lysandre Debut Committed by GitHub
Browse files

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

parent 3290315a
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
tests/test_modeling_megatron_bert.py
View file @ 8c7bd1b9
...@@ -356,6 +356,7 @@ TOLERANCE = 1e-4 ...@@ -356,6 +356,7 @@ TOLERANCE = 1e-4
@require_tokenizers @require_tokenizers
class MegatronBertModelIntegrationTests(unittest.TestCase): class MegatronBertModelIntegrationTests(unittest.TestCase):
@slow @slow
@unittest.skip("Model is not available.")
def test_inference_no_head(self): def test_inference_no_head(self):
directory = "nvidia/megatron-bert-uncased-345m" directory = "nvidia/megatron-bert-uncased-345m"
if "MYDIR" in os.environ: if "MYDIR" in os.environ:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment