Switch to bert model for testing.

SamanehSaadat · Apr 16, 2024 · 02be2bf · 02be2bf
1 parent a8845ea
commit 02be2bf
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/keras_nlp/models/backbone_test.py b/keras_nlp/models/backbone_test.py
@@ -44,4 +44,4 @@ def test_from_preset_errors(self):
             GPT2Backbone.from_preset("bert_tiny_en_uncased", load_weights=False)
         with self.assertRaises(FileNotFoundError):
             # No loading on a non-keras model.
-            Backbone.from_preset("hf://google/gemma-2b")
+            Backbone.from_preset("hf://google-bert/bert-base-uncased")
diff --git a/keras_nlp/models/preprocessor_test.py b/keras_nlp/models/preprocessor_test.py
@@ -51,6 +51,6 @@ def test_from_preset_errors(self):
             BertPreprocessor.from_preset("gpt2_base_en")
         with self.assertRaises(FileNotFoundError):
             # No loading on a non-keras model.
-            Preprocessor.from_preset("hf://google/gemma-2b")
+            Preprocessor.from_preset("hf://google-bert/bert-base-uncased")
 
     # TODO: Add more tests when we added a model that has `preprocessor.json`.
diff --git a/keras_nlp/models/task_test.py b/keras_nlp/models/task_test.py
@@ -71,7 +71,7 @@ def test_from_preset_errors(self):
             BertClassifier.from_preset("gpt2_base_en", load_weights=False)
         with self.assertRaises(FileNotFoundError):
             # No loading on a non-keras model.
-            CausalLM.from_preset("hf://google/gemma-2b")
+            CausalLM.from_preset("hf://google-bert/bert-base-uncased")
 
     def test_summary_with_preprocessor(self):
         preprocessor = SimplePreprocessor()

diff --git a/keras_nlp/tokenizers/tokenizer_test.py b/keras_nlp/tokenizers/tokenizer_test.py
@@ -56,7 +56,7 @@ def test_from_preset_errors(self):
             GPT2Tokenizer.from_preset("bert_tiny_en_uncased")
         with self.assertRaises(FileNotFoundError):
             # No loading on a non-keras model.
-            Tokenizer.from_preset("hf://google/gemma-2b")
+            Tokenizer.from_preset("hf://google-bert/bert-base-uncased")
 
     def test_tokenize(self):
         input_data = ["the quick brown fox"]