saymrwulf/transformers
1
0
Fork 0
mirror of https://github.com/saymrwulf/transformers.git synced 2026-05-14 20:58:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

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