fix

huggingface · Jul 4, 2024 · bf85e35 · bf85e35
1 parent d0a1c31
commit bf85e35
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/models/gemma/test_modeling_gemma.py b/tests/models/gemma/test_modeling_gemma.py
@@ -542,7 +542,7 @@ def setUpClass(cls):
 
     @require_read_token
     def test_model_2b_fp16(self):
-        model_id = "google/gemma-2-9b"
+        model_id = "google/gemma-2b"
         EXPECTED_TEXTS = [
             "Hello I am doing a project on the 1990s and I need to know what the most popular music",
             "Hi today I am going to share with you a very easy and simple recipe of <strong><em>Kaju Kat",