transformers/tests/quantization
Jerry Zhang 44fa04ae8d
Include output embedding as well with include_embedding flag (#37935)
* Include output embedding as well with `include_embedding` flag

Summary:
att

Test Plan:
python tests/quantization/torchao_integration/test_torchao.py -k test_include_embedding

Reviewers:

Subscribers:

Tasks:

Tags:

* format

* rename include_embedding to include_input_output_embeddings

---------

Co-authored-by: Mohamed Mekkouri <93391238+MekkCyber@users.noreply.github.com>
2025-05-16 12:06:11 +02:00
..
aqlm_integration Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +02:00
autoawq Fixing quantization tests (#37650) 2025-04-22 13:59:57 +02:00
autoround enable autoround cases on XPU (#38167) 2025-05-16 09:08:35 +00:00
bitnet_integration Add Bitnet model (#37742) 2025-04-28 15:08:46 +02:00
bnb update bnb tests (#38011) 2025-05-08 20:35:24 +00:00
compressed_tensors_integration Fix: Unexpected Keys, Improve run_compressed, Rename Test Folder (#37077) 2025-04-04 21:30:11 +02:00
eetq_integration Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +02:00
fbgemm_fp8 Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +02:00
finegrained_fp8 enable finegrained_fp8 and granite_speech cases on XPU (#38036) 2025-05-14 08:58:40 +00:00
ggml Fixing quantization tests (#37650) 2025-04-22 13:59:57 +02:00
gptq Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +02:00
higgs Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +02:00
hqq Fixes hqq by following a new path for bias parameter in pre_quantized models (#37530) 2025-04-16 13:58:14 +02:00
quanto_integration Fixing quantization tests (#37650) 2025-04-22 13:59:57 +02:00
quark_integration Fix typos in strings and comments (#37799) 2025-04-28 11:39:11 +01:00
spqr_integration Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +02:00
torchao_integration Include output embedding as well with include_embedding flag (#37935) 2025-05-16 12:06:11 +02:00
vptq_integration Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +02:00