TF: tests for (de)serializable models with resized tokens (#19013)

* resized models that we can actually load

* separate embeddings check

* add test for embeddings out of bounds

* add fake slows
This commit is contained in:
Joao Gante
2022-09-16 16:38:08 +01:00
committed by GitHub
parent 70ba10e6d4
commit 658010c739
4 changed files with 123 additions and 4 deletions

View File

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