File tree Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change 1
- 711fa0809f06fc97febd0c3fe72563c3fe227e51
1
+ 7513042f39515af4c643bc1f9399952ad7f4f904
Original file line number Diff line number Diff line change 55
55
)
56
56
from torchao .experimental .quant_api import (
57
57
int8_dynamic_activation_intx_weight ,
58
- IntxWeightEmbeddingQuantizer ,
58
+ EmbeddingQuantizer ,
59
59
)
60
60
from torchao .quantization .granularity import (
61
61
PerGroup ,
@@ -939,7 +939,7 @@ def quantized_model(self) -> nn.Module:
939
939
# class references
940
940
quantizer_class_dict = {
941
941
"embedding" : EmbeddingOnlyQuantHandler ,
942
- "embedding:wx" : IntxWeightEmbeddingQuantizer ,
942
+ "embedding:wx" : EmbeddingQuantizer ,
943
943
"linear:int8" : WeightOnlyInt8QuantHandler ,
944
944
"precision" : PrecisionHandler ,
945
945
"executor" : ExecutorHandler ,
You can’t perform that action at this time.
0 commit comments