up

metascroy · metascroy · commit 37b256461739 · 2025-05-22T10:12:17.000-07:00
diff --git a/extension/llm/export/quantizer_lib.py b/extension/llm/export/quantizer_lib.py
@@ -16,7 +16,8 @@
     XNNPACKQuantizer,
 )
 
-from torchao.quantization.pt2e.quantizer import EmbeddingQuantizer, Quantizer
+from torchao.quantization.pt2e.quantizer import Quantizer
+from torchao.quantization.pt2e.quantizer.embedding_quantizer import EmbeddingQuantizer
 
 FORMAT = "[%(levelname)s %(asctime)s %(filename)s:%(lineno)s] %(message)s"
 logging.basicConfig(level=logging.INFO, format=FORMAT)

Original file line number	Diff line number	Diff line change
`@@ -16,7 +16,8 @@`
`16`	`16`	`XNNPACKQuantizer,`
`17`	`17`	`)`
`18`	`18`
`19`		`-from torchao.quantization.pt2e.quantizer import EmbeddingQuantizer, Quantizer`
	`19`	`+from torchao.quantization.pt2e.quantizer import Quantizer`
	`20`	`+from torchao.quantization.pt2e.quantizer.embedding_quantizer import EmbeddingQuantizer`
`20`	`21`
`21`	`22`	`FORMAT = "[%(levelname)s %(asctime)s %(filename)s:%(lineno)s] %(message)s"`
`22`	`23`	`logging.basicConfig(level=logging.INFO, format=FORMAT)`