Skip to content

Commit

Permalink
fix: Fixed failing test_find_base_model_checkpoint (#32638)
Browse files Browse the repository at this point in the history
Fixed failing test_find_base_model_checkpoint.
  • Loading branch information
Sai-Suraj-27 authored Aug 12, 2024
1 parent 7f777ab commit ce4b288
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/utils/test_add_new_model_like.py
Original file line number Diff line number Diff line change
Expand Up @@ -657,7 +657,7 @@ def test_get_model_files_tf_and_flax(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("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()}
Expand Down

0 comments on commit ce4b288

Please sign in to comment.