From ce4b28830a439b439e440e8b0025c2fdfceb5927 Mon Sep 17 00:00:00 2001 From: Sai-Suraj-27 Date: Tue, 13 Aug 2024 00:21:30 +0530 Subject: [PATCH] fix: Fixed failing `test_find_base_model_checkpoint` (#32638) Fixed failing test_find_base_model_checkpoint. --- tests/utils/test_add_new_model_like.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/utils/test_add_new_model_like.py b/tests/utils/test_add_new_model_like.py index 9c150b32bd..1eb6d56e6c 100644 --- a/tests/utils/test_add_new_model_like.py +++ b/tests/utils/test_add_new_model_like.py @@ -657,7 +657,7 @@ NEW_BERT_CONSTANT = "value" def test_find_base_model_checkpoint(self): self.assertEqual(find_base_model_checkpoint("bert"), "google-bert/bert-base-uncased") - self.assertEqual(find_base_model_checkpoint("gpt2"), "gpt2") + self.assertEqual(find_base_model_checkpoint("gpt2"), "openai-community/gpt2") def test_retrieve_model_classes(self): gpt_classes = {k: set(v) for k, v in retrieve_model_classes("gpt2").items()}