Skip to content

GitLab

"docs/source/vscode:/vscode.git/clone" did not exist on "5bb4430edc7df9f9950d412d98bbe505cc4d328b"
Unverified Commit c8849583 authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Make `test_save_pretrained_signatures` slow test (#21105) 


Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent 14154f72
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
tests/test_modeling_tf_common.py
View file @ c8849583
...@@ -2266,6 +2266,7 @@ class UtilsFunctionsTest(unittest.TestCase): ...@@ -2266,6 +2266,7 @@ class UtilsFunctionsTest(unittest.TestCase):
for p1, p2 in zip(model.weights, new_model.weights): for p1, p2 in zip(model.weights, new_model.weights):
self.assertTrue(np.allclose(p1.numpy(), p2.numpy())) self.assertTrue(np.allclose(p1.numpy(), p2.numpy()))
@slow
def test_save_pretrained_signatures(self): def test_save_pretrained_signatures(self):
model = TFBertModel.from_pretrained("hf-internal-testing/tiny-random-bert") model = TFBertModel.from_pretrained("hf-internal-testing/tiny-random-bert")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment