Skip to content

Commit

Permalink
Update on "Update QAT READMEs using new APIs"
Browse files Browse the repository at this point in the history
Add references to new QAT APIs including `quantize_`,
`FakeQuantizedX`, and the new embedding Quantizers and
ComposableQATQuantizer. Also link to new QAT + LoRA recipe
in torchtune.

[ghstack-poisoned]
  • Loading branch information
andrewor14 committed Jan 10, 2025
2 parents 50a8355 + ec94797 commit fd55517
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion torchao/quantization/qat/__init__.py
Original file line number Diff line number Diff line change
@@ -1,8 +1,8 @@
from .api import (
ComposableQATQuantizer,
FakeQuantizeConfig,
intx_quantization_aware_training,
from_intx_quantization_aware_training,
intx_quantization_aware_training,
)
from .embedding import (
Int4WeightOnlyEmbeddingQATQuantizer,
Expand Down

0 comments on commit fd55517

Please sign in to comment.