transformers/tests/quantization/torchao_integration
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
..
__init__.py Add TorchAOHfQuantizer (#32306) 2024-08-14 16:14:24 +02:00
test_torchao.py Include output embedding as well with include_embedding flag (#37935) 2025-05-16 12:06:11 +02:00