From 4c99cc720e12800c5f11ac52f4cc325583a15e1f Mon Sep 17 00:00:00 2001 From: matt Date: Tue, 14 Dec 2021 19:29:33 +0000 Subject: [PATCH] Fix accidental edit to test_modeling_common --- tests/test_modeling_common.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/test_modeling_common.py b/tests/test_modeling_common.py index b58d90aacd1b..6640028293e2 100755 --- a/tests/test_modeling_common.py +++ b/tests/test_modeling_common.py @@ -2193,6 +2193,7 @@ def test_push_to_hub(self): model = BertModel(config) with tempfile.TemporaryDirectory() as tmp_dir: model.save_pretrained(os.path.join(tmp_dir, "test-model"), push_to_hub=True, use_auth_token=self._token) + new_model = BertModel.from_pretrained(f"{USER}/test-model") for p1, p2 in zip(model.parameters(), new_model.parameters()): self.assertTrue(torch.equal(p1, p2))