Swap gated meta-llama/llama-3.2 with allenai/llama (#10)

huggingface · Dec 21, 2024 · a529795 · a529795
1 parent e43dba8
commit a529795
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/generation/test_candidate_generator.py b/tests/generation/test_candidate_generator.py
@@ -268,7 +268,7 @@ def setUpClass(cls):
         cls.assistant_model = AutoModelForCausalLM.from_pretrained("hf-internal-testing/tiny-random-gpt2").to(
             cls.device
         )
-        cls.main_tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-3.2-1B-Instruct")
+        cls.main_tokenizer = AutoTokenizer.from_pretrained("allenai/Llama-3.1-Tulu-3-8B-SFT")
         cls.assistant_tokenizer = AutoTokenizer.from_pretrained("hf-internal-testing/tiny-random-gpt2")
         cls.generation_config = GenerationConfig()