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

fix: Fixed failing test_find_base_model_checkpoint (#32638)

Browse Source 
Fixed failing test_find_base_model_checkpoint.
This commit is contained in:
Sai-Suraj-27
2024-08-13 00:21:30 +05:30
committed by GitHub
parent 7f777ab7d9
commit ce4b28830a
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/utils/test_add_new_model_like.py
View File

@@ -657,7 +657,7 @@ NEW_BERT_CONSTANT = "value"
def test_find_base_model_checkpoint(self): 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("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): def test_retrieve_model_classes(self):
gpt_classes = {k: set(v) for k, v in retrieve_model_classes("gpt2").items()} gpt_classes = {k: set(v) for k, v in retrieve_model_classes("gpt2").items()}