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

small test fix
2024-11-22 17:18:49 +01:00
parent 286ffaaf0a
commit 54be2d7ae8
1 changed files with 1 additions and 1 deletions
--- a/tests/quantization/bitnet_integration/test_bitnet.py
+++ b/tests/quantization/bitnet_integration/test_bitnet.py
@@ -65,7 +65,7 @@ class BitNetTest(unittest.TestCase):
        """
        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):