SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 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
1 changed files with 1 additions and 1 deletions
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 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) cls.quantized_model = AutoModelForCausalLM.from_pretrained(cls.model_name, device_map=cls.device)
def tearDown(self): def tearDown(self):