From e1224a2a0f42c2d00f5b9d93d4d5dd43a628b574 Mon Sep 17 00:00:00 2001 From: Sylvain Gugger Date: Wed, 14 Sep 2022 10:01:22 -0400 Subject: [PATCH] Making save_load test slow as it times out --- tests/models/vit_mae/test_modeling_tf_vit_mae.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/models/vit_mae/test_modeling_tf_vit_mae.py b/tests/models/vit_mae/test_modeling_tf_vit_mae.py index 906c79e766..99f851d2a9 100644 --- a/tests/models/vit_mae/test_modeling_tf_vit_mae.py +++ b/tests/models/vit_mae/test_modeling_tf_vit_mae.py @@ -328,6 +328,7 @@ class TFViTMAEModelTest(TFModelTesterMixin, unittest.TestCase): # overwrite from common since TFViTMAEForPretraining has random masking, we need to fix the noise # to generate masks during test + @slow() def test_keras_save_load(self): # make mask reproducible np.random.seed(2)