From 4eb36f2921fed7d57aa9ff27a05942bd9402c6f0 Mon Sep 17 00:00:00 2001
From: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
Date: Wed, 14 Sep 2022 10:38:39 -0400
Subject: [PATCH] Mark right save_load test as slow (#19031)

---
 tests/models/vit_mae/test_modeling_tf_vit_mae.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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 e0613b3b4b..e9db7ea6b2 100644
--- a/tests/models/vit_mae/test_modeling_tf_vit_mae.py
+++ b/tests/models/vit_mae/test_modeling_tf_vit_mae.py
@@ -328,7 +328,6 @@ 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)
@@ -376,6 +375,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_save_load(self):
         # make mask reproducible
         np.random.seed(2)