riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-02 19:21:31 +06:00
Code Issues Actions 3 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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats 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):
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()}