SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Lysandre Debut
2021-07-15 15:14:12 +02:00
committed by GitHub
parent 8c7bd1b97b
commit eb2e006b35
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tests/test_modeling_megatron_gpt2.py
View File

@@ -31,6 +31,7 @@ if is_torch_available():
@require_tokenizers @require_tokenizers
class MegatronGPT2IntegrationTest(unittest.TestCase): class MegatronGPT2IntegrationTest(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-gpt2-345m/" directory = "nvidia/megatron-gpt2-345m/"
if "MYDIR" in os.environ: if "MYDIR" in os.environ: