Bitnet test fix to avoid using gated model (#34863)

small test fix
huggingface · Nov 22, 2024 · 54be2d7 · 54be2d7
1 parent 286ffaa
commit 54be2d7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/quantization/bitnet_integration/test_bitnet.py b/tests/quantization/bitnet_integration/test_bitnet.py
@@ -65,7 +65,7 @@ def setUpClass(cls):
         """
         Load the model
         """
-        cls.tokenizer = AutoTokenizer.from_pretrained("meta-llama/Meta-Llama-3-8B-Instruct")
+        cls.tokenizer = AutoTokenizer.from_pretrained(cls.model_name)
         cls.quantized_model = AutoModelForCausalLM.from_pretrained(cls.model_name, device_map=cls.device)
 
     def tearDown(self):