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

Fix typo in EETQ Tests (#35160)

Browse source 
fix
This commit is contained in:
Mohamed Mekkouri 2024-12-09 14:13:36 +01:00 committed by GitHub
parent de8a0b7547
commit 7238387f67
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
tests/quantization/eetq_integration/test_eetq.py
View file

@ -119,7 +119,7 @@ class EetqTest(unittest.TestCase):
self.assertEqual(nb_linears - 1, nb_eetq_linear)
# Try with `linear_weights_not_to_quantize`
# Try with `modules_to_not_convert`
with init_empty_weights():
model = OPTForCausalLM(config)
quantization_config = EetqConfig(modules_to_not_convert=["fc1"])
@ -128,7 +128,7 @@ class EetqTest(unittest.TestCase):
for module in model.modules():
if isinstance(module, EetqLinear):
nb_eetq_linear += 1
# 25 corresponds to the lm_head along with 24 fc1 layers.
self.assertEqual(nb_linears - 25, nb_eetq_linear)
def test_quantized_model(self):