riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

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

Browse Source 
small test fix
This commit is contained in:
Mohamed Mekkouri 2024-11-22 17:18:49 +01:00 committed by GitHub
parent 286ffaaf0a
commit 54be2d7ae8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/quantization/bitnet_integration/test_bitnet.py
View File

@ -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):