We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 6a1d4c5 commit 155f392Copy full SHA for 155f392
torchao/experimental/quant_api.py
@@ -499,8 +499,8 @@ def quantize(self, model: nn.Module) -> nn.Module:
499
)
500
from torchao.quantization.quant_api import (
501
MappingType,
502
- to_affine_quantized_intx,
503
ZeroPointDomain,
+ to_affine_quantized_intx,
504
505
from torchao.quantization.transform_module import register_quantize_module_handler
506
from torchao.quantization.utils import _get_per_token_block_size
0 commit comments