Skip test while the model is not available (#12740)
This commit is contained in:
parent
8c7bd1b97b
commit
eb2e006b35
|
@ -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:
|
||||
|
|
Loading…
Reference in New Issue