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

Mark right save_load test as slow (#19031)

Browse Source
This commit is contained in:
Sylvain Gugger
2022-09-14 10:38:39 -04:00
committed by GitHub
parent f5f430e5c8
commit 4eb36f2921
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/models/vit_mae/test_modeling_tf_vit_mae.py
View File

@@ -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)