Skip to content

GitLab

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

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

parent 8c7bd1b9
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
tests/test_modeling_megatron_gpt2.py
View file @ eb2e006b
......@@ -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:
......
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