fixed gemma tokenizer test

mmoffatt2 · Dec 10, 2024 · 1b7d4a2 · 1b7d4a2
1 parent b1e744e
commit 1b7d4a2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/data/template/tests.py b/data/template/tests.py
@@ -210,7 +210,7 @@ def test_custom_char_tokenizer_with_byte_fallback(self):
             os.remove(args.custom_chars_file)
 
     def test_gemma_tokenizer(self):
-        args = Namespace(qwen2_model="gemma-2b")
+        args = Namespace(gemma_model="gemma-2b")
         tokenizer = GemmaTokenizer(args)
 
         # Tokenize